@@ -125,19 +125,19 @@ def _test(sd):
125125 _test ({"max_epochs" : 100 , "epoch_length" : 120 , "iteration" : 123 })
126126 _test ({"max_epochs" : 100 , "epoch_length" : 120 , "epoch" : 5 })
127127
128- with pytest .raises (ValueError , match = r"Argument max_epochs should be larger than" ):
128+ with pytest .raises (ValueError , match = r"Argument max_epochs should be greater than the current " ):
129129 _test ({"max_epochs" : 10 , "epoch_length" : 120 , "epoch" : 50 })
130130
131- with pytest .raises (ValueError , match = r"Argument max_epochs should be larger than" ):
131+ with pytest .raises (ValueError , match = r"Argument max_epochs should be greater than the current " ):
132132 _test ({"max_epochs" : 10 , "epoch_length" : 120 , "iteration" : 5000 })
133133
134134 _test ({"max_iters" : 500 , "epoch_length" : 120 , "iteration" : 123 })
135135 _test ({"max_iters" : 500 , "epoch_length" : 120 , "epoch" : 3 })
136136
137- with pytest .raises (ValueError , match = r"Argument max_iters should be larger than" ):
137+ with pytest .raises (ValueError , match = r"Argument max_iters should be greater than" ):
138138 _test ({"max_iters" : 500 , "epoch_length" : 120 , "epoch" : 5 })
139139
140- with pytest .raises (ValueError , match = r"Argument max_iters should be larger than" ):
140+ with pytest .raises (ValueError , match = r"Argument max_iters should be greater than" ):
141141 _test ({"max_iters" : 500 , "epoch_length" : 120 , "iteration" : 501 })
142142
143143
@@ -184,7 +184,7 @@ def test_load_state_dict_with_params_overriding_integration():
184184 assert state .max_epochs == new_max_epochs
185185 assert state .iteration == state_dict ["epoch_length" ] * new_max_epochs
186186 assert state .epoch == new_max_epochs
187- with pytest .raises (ValueError , match = r"Argument max_epochs should be larger than the current epoch" ):
187+ with pytest .raises (ValueError , match = r"Argument max_epochs should be greater than the current epoch" ):
188188 engine .load_state_dict (state_dict )
189189 engine .run (data , max_epochs = 3 )
190190
0 commit comments