simplify by removing duplicate base_model_config (#772)

2023-10-23 01:42:38 -04:00
parent 44c9d0151a
commit 2d8def68dc
39 changed files with 10 additions and 42 deletions
--- a/tests/test_normalize_config.py
+++ b/tests/test_normalize_config.py
@@ -37,3 +37,10 @@ class NormalizeConfigTestCase(unittest.TestCase):
        normalize_config(cfg)

        assert cfg.learning_rate == 0.00005
+
+    def test_base_model_config_set_when_empty(self):
+        cfg = self._get_base_cfg()
+        del cfg.base_model_config
+        normalize_config(cfg)
+
+        assert cfg.base_model_config == cfg.base_model