From ad48193c1c1329ce78b5a0501d34a68441b694ac Mon Sep 17 00:00:00 2001 From: Padmanabha V Seshadri Date: Mon, 22 Jul 2024 01:43:24 -0400 Subject: [PATCH] fix: Format issues resolved Signed-off-by: Padmanabha V Seshadri --- tests/data/trainercontroller/__init__.py | 4 +--- tests/trainercontroller/test_tuning_trainercontroller.py | 5 ++--- 2 files changed, 3 insertions(+), 6 deletions(-) diff --git a/tests/data/trainercontroller/__init__.py b/tests/data/trainercontroller/__init__.py index a6910dbbf..35f4f13c9 100644 --- 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 index 1eed0e6ba..c4464da89 100644 --- 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(