From e56ee7500bcb8106364aca389dc2a07d5fe6a568 Mon Sep 17 00:00:00 2001 From: Vassilis Vassiliadis Date: Wed, 26 Jun 2024 08:53:41 +0100 Subject: [PATCH] refactor: update test_run_with_additional_callbacks() unit-test Signed-off-by: Vassilis Vassiliadis --- tests/test_sft_trainer.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tests/test_sft_trainer.py b/tests/test_sft_trainer.py index 085e8da0b..ed7b21d97 100644 --- a/tests/test_sft_trainer.py +++ b/tests/test_sft_trainer.py @@ -629,8 +629,7 @@ def test_run_with_additional_callbacks(): """Ensure that train() can work with additional_callbacks""" with tempfile.TemporaryDirectory() as tempdir: - train_args = copy.deepcopy(TRAIN_ARGS) - train_args.output_dir = tempdir + train_args = get_train_args(output_dir=tempdir) model_args = copy.deepcopy(MODEL_ARGS) sft_trainer.train(