Skip to content

Commit

Permalink
Remove redundant test
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 18, 2024
1 parent 6a02e1b commit 5b89073
Showing 1 changed file with 0 additions and 13 deletions.
13 changes: 0 additions & 13 deletions tests/test_sft_trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -409,19 +409,6 @@ def test_empty_data():
sft_trainer.train(model_args, data_args, training_args, tune_config)


def test_data_path_does_not_exist():
"""Ensure that we get a FileNotFoundError if the data is missing completely."""
TRAIN_KWARGS = {
**BASE_PEFT_KWARGS,
**{"training_data_path": "/foo/bar/foobar", "output_dir": "foo/bar/baz"},
}
model_args, data_args, training_args, tune_config = causal_lm_train_kwargs(
TRAIN_KWARGS
)
with pytest.raises(FileNotFoundError):
sft_trainer.train(model_args, data_args, training_args, tune_config)


def test_data_path_is_a_directory():
"""Ensure that we get FileNotFoundError if we point the data path at a dir, not a file."""
with tempfile.TemporaryDirectory() as tempdir:
Expand Down

0 comments on commit 5b89073

Please sign in to comment.