Modify test file expected_optimizer

sifis-home · Oct 4, 2023 · 5f2c3a5 · 5f2c3a5
1 parent 997c07c
commit 5f2c3a5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/test_flask-parental-control.py b/test_flask-parental-control.py
@@ -138,6 +138,7 @@ def test_get_optimizer():
 
     if cfg.train.optimizer_name == "adam":
         expected_optimizer = Adam(lr=cfg.train.lr)
+        expected_optimizer2 = load_optimizer(cfg)
     else:
         raise ValueError("optimizer name should be 'adam'")
 
@@ -183,7 +184,7 @@ def test_load_optimizer_sgd_none():
         ]
     )
 
-    optimizer = load_optimizer(cfg)
+    expected_optimizer = load_optimizer(cfg)
 
     optimizer_name = "Other"
     cfg = OmegaConf.from_dotlist(
@@ -196,6 +197,7 @@ def test_load_optimizer_sgd_none():
             f"train.batch_size={batch_size}",
         ]
     )
+    expected_optimizer = load_optimizer(cfg)
 
 
 def test_get_optimizer():