From 146e9f144aac533966e2759b8c456cf1ec757aff Mon Sep 17 00:00:00 2001 From: Sukriti-Sharma4 Date: Thu, 19 Sep 2024 11:39:11 -0600 Subject: [PATCH] fix: unit tests Signed-off-by: Sukriti-Sharma4 --- tests/test_sft_trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_sft_trainer.py b/tests/test_sft_trainer.py index b20547002..d243727e0 100644 --- a/tests/test_sft_trainer.py +++ b/tests/test_sft_trainer.py @@ -692,7 +692,7 @@ def test_run_causallm_ft_save_with_save_model_dir_save_strategy_no(): save_model_args.save_strategy = "no" save_model_args.output_dir = tempdir - trainer = sft_trainer.train(MODEL_ARGS, DATA_ARGS, save_model_args, None) + trainer, _ = sft_trainer.train(MODEL_ARGS, DATA_ARGS, save_model_args, None) logs_path = os.path.join( tempdir, FileLoggingTrackerConfig.training_logs_filename )