From b0f745085604519542d4721bd4b31d51b3d7475f Mon Sep 17 00:00:00 2001 From: Alex-Brooks Date: Wed, 24 Apr 2024 10:28:35 -0600 Subject: [PATCH] Fix linting Signed-off-by: Alex-Brooks --- tests/test_sft_trainer.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/test_sft_trainer.py b/tests/test_sft_trainer.py index 11157bfba..c23c7e2c5 100644 --- a/tests/test_sft_trainer.py +++ b/tests/test_sft_trainer.py @@ -419,8 +419,9 @@ def test_data_path_is_a_directory(): model_args, data_args, training_args, tune_config = causal_lm_train_kwargs( TRAIN_KWARGS ) - # Confusingly, if we pass a directory for our data path, it will throw a FileNotFoundError saying - # "unable to find ''", since it can't find a matchable file in the path. + # Confusingly, if we pass a directory for our data path, it will throw a + # FileNotFoundError saying "unable to find ''", since it can't + # find a matchable file in the path. with pytest.raises(FileNotFoundError): sft_trainer.train(model_args, data_args, training_args, tune_config)