Skip to content

Commit

Permalink
Fix linting
Browse files Browse the repository at this point in the history
Signed-off-by: Alex-Brooks <[email protected]>
  • Loading branch information
alex-jw-brooks committed Apr 24, 2024
1 parent 5b89073 commit b0f7450
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions tests/test_sft_trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -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 '<data_path>'", 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 '<data_path>'", 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)

Expand Down

0 comments on commit b0f7450

Please sign in to comment.