Update test_validation.py

axolotl-ai-cloud · Apr 3, 2024 · 595ea1d · 595ea1d
1 parent 6a634e2
commit 595ea1d
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/tests/test_validation.py b/tests/test_validation.py
@@ -1060,7 +1060,9 @@ def test_hub_model_id_save_value_warns_save_stragey_no(self, minimal_cfg):
             )
 
     def test_hub_model_id_save_value_warns_random_value(self, minimal_cfg):
-        cfg = DictDefault({"hub_model_id": "test", "save_strategy": "test"}) | minimal_cfg
+        cfg = (
+            DictDefault({"hub_model_id": "test", "save_strategy": "test"}) | minimal_cfg
+        )
 
         with self._caplog.at_level(logging.WARNING):
             validate_config(cfg)
@@ -1069,14 +1071,20 @@ def test_hub_model_id_save_value_warns_random_value(self, minimal_cfg):
             )
 
     def test_hub_model_id_save_value_steps(self, minimal_cfg):
-        cfg = DictDefault({"hub_model_id": "test", "save_strategy": "steps"}) | minimal_cfg
+        cfg = (
+            DictDefault({"hub_model_id": "test", "save_strategy": "steps"})
+            | minimal_cfg
+        )
 
         with self._caplog.at_level(logging.WARNING):
             validate_config(cfg)
             assert len(self._caplog.records) == 0
 
     def test_hub_model_id_save_value_epochs(self, minimal_cfg):
-        cfg = DictDefault({"hub_model_id": "test", "save_strategy": "epoch"}) | minimal_cfg
+        cfg = (
+            DictDefault({"hub_model_id": "test", "save_strategy": "epoch"})
+            | minimal_cfg
+        )
 
         with self._caplog.at_level(logging.WARNING):
             validate_config(cfg)