Skip to content

Commit 3c5146f

Browse files
committed
fix: update validation message for swa_start in TrainingConfig
1 parent a4ba973 commit 3c5146f

1 file changed

Lines changed: 1 addition & 5 deletions

File tree

tests/training/test_train_swa.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -501,11 +501,7 @@ def test_training_config_validation():
501501
with pytest.raises(ValueError, match='n_epoch must be positive'):
502502
TrainingConfig(n_epoch=0)
503503

504-
# Invalid: swa_start must be between 0 and n_epoch
505-
with pytest.raises(ValueError, match='swa_start must be between'):
506-
TrainingConfig(n_epoch=100, swa_start=100)
507-
508-
with pytest.raises(ValueError, match='swa_start must be between'):
504+
with pytest.raises(ValueError, match='swa_start must be non-negative'):
509505
TrainingConfig(n_epoch=100, swa_start=-1)
510506

511507
# Invalid: swa_freq must be positive

0 commit comments

Comments
 (0)