diff --git a/tests/e2e/test_falcon.py b/tests/e2e/test_falcon.py index 21e23ca11e..c76699a7c8 100644 --- a/tests/e2e/test_falcon.py +++ b/tests/e2e/test_falcon.py @@ -71,7 +71,7 @@ def test_lora(self, temp_dir): dataset_meta = load_datasets(cfg=cfg, cli_args=cli_args) train(cfg=cfg, cli_args=cli_args, dataset_meta=dataset_meta) - assert (Path(temp_dir) / "adapter_model.bin").exists() @ with_temp_dir + assert (Path(temp_dir) / "adapter_model.bin").exists() @with_temp_dir def test_lora_added_vocab(self, temp_dir):