Update test_sr_model.py

the-database · Oct 6, 2024 · bb5113c · bb5113c
1 parent e8ac6f6
commit bb5113c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/test_models/test_sr_model.py b/tests/test_models/test_sr_model.py
@@ -25,6 +25,9 @@ def test_srmodel(monkeypatch: MonkeyPatch) -> None:
     opt.use_amp = False
     opt.path.pretrain_network_g = None
     opt.path.pretrain_network_g_ema = None
+    assert opt.val is not None
+    opt.val.val_enabled = True
+    opt.val.metrics_enabled = True
 
     # build model
     model = SRModel(opt)