From 595ea1dd5b1cf27bc18c3949cfee3393301ba0f4 Mon Sep 17 00:00:00 2001
From: Johan Hansson <39947546+JohanWork@users.noreply.github.com>
Date: Wed, 3 Apr 2024 16:36:42 +0200
Subject: [PATCH] Update test_validation.py

---
 tests/test_validation.py | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

diff --git a/tests/test_validation.py b/tests/test_validation.py
index a7c8a22bb4..9ab9ed0712 100644
--- 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)