From 5f2c3a5e0c9f0f727f861284d3c527a1c50031c7 Mon Sep 17 00:00:00 2001 From: Wisam Abbasi Date: Wed, 4 Oct 2023 10:50:01 +0200 Subject: [PATCH] Modify test file expected_optimizer --- test_flask-parental-control.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/test_flask-parental-control.py b/test_flask-parental-control.py index a47eb23..b7f3b46 100644 --- 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():