fix: Format issues resolved

Signed-off-by: Padmanabha V Seshadri <[email protected]>
foundation-model-stack · Jul 30, 2024 · ad48193 · ad48193
1 parent 69db9d1
commit ad48193
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 6 deletions.
diff --git a/tests/data/trainercontroller/__init__.py b/tests/data/trainercontroller/__init__.py
@@ -77,6 +77,4 @@
 TRAINER_CONFIG_TEST_THRESHOLDED_TRAINING_LOSS_YAML = os.path.join(
     _DATA_DIR, "thresholded-training-loss.yaml"
 )
-TRAINER_CONFIG_TEST_ON_SAVE_YAML = os.path.join(
-    _DATA_DIR, "on-save.yaml"
-)
+TRAINER_CONFIG_TEST_ON_SAVE_YAML = os.path.join(_DATA_DIR, "on-save.yaml")
diff --git a/tests/trainercontroller/test_tuning_trainercontroller.py b/tests/trainercontroller/test_tuning_trainercontroller.py
@@ -137,14 +137,13 @@ def test_thresholded_training_loss():
     tc_callback.on_log(args=test_data.args, state=test_data.states[2], control=control)
     assert control.should_training_stop is True
 
+
 def test_thresholded_training_loss_on_save():
     """Tests the thresholded training loss example in
     `examples/trainer-controller-configs/on-save.yaml`
     """
     test_data = _setup_data()
-    tc_callback = tc.TrainerControllerCallback(
-        td.TRAINER_CONFIG_TEST_ON_SAVE_YAML
-    )
+    tc_callback = tc.TrainerControllerCallback(td.TRAINER_CONFIG_TEST_ON_SAVE_YAML)
     control = TrainerControl(should_training_stop=False)
     # Trigger on_init_end to perform registration of handlers to events
     tc_callback.on_init_end(