Test whether the warning is issued

Noble-Lab · Feb 21, 2024 · a950ee6 · a950ee6
1 parent be89749
commit a950ee6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/unit_tests/test_runner.py b/tests/unit_tests/test_runner.py
@@ -122,7 +122,8 @@ def test_save_and_load_weights_deprecated(tmp_path, mgf_small, tiny_config):
         assert runner.model.cosine_schedule_period_iters == 5
     # Fine-tuning.
     with ModelRunner(config=config, model_filename=str(ckpt)) as runner:
-        runner.train([mgf_small], [mgf_small])
+        with pytest.warns(DeprecationWarning):
+            runner.train([mgf_small], [mgf_small])
 
 
 def test_calculate_precision(tmp_path, mgf_small, tiny_config):