From 4935b678bfdc0831b7b7c6e9614a9b37ae6eadb8 Mon Sep 17 00:00:00 2001 From: ankona <3595025+ankona@users.noreply.github.com> Date: Mon, 29 Jul 2024 12:30:06 -0500 Subject: [PATCH] fix test failing new validation check --- tests/mli/test_core_machine_learning_worker.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/mli/test_core_machine_learning_worker.py b/tests/mli/test_core_machine_learning_worker.py index c7e1cb2863..6fa9f9944e 100644 --- a/tests/mli/test_core_machine_learning_worker.py +++ b/tests/mli/test_core_machine_learning_worker.py @@ -85,12 +85,12 @@ def persist_torch_tensor(test_dir: str) -> pathlib.Path: @pytest.mark.skipif(not torch_available, reason="Torch backend is not installed") -def test_fetch_model_disk(persist_torch_model: pathlib.Path) -> None: +def test_fetch_model_disk(persist_torch_model: pathlib.Path, test_dir: str) -> None: """Verify that the ML worker successfully retrieves a model when given a valid (file system) key""" worker = MachineLearningWorkerCore key = str(persist_torch_model) - feature_store = FileSystemFeatureStore() + feature_store = FileSystemFeatureStore(test_dir) fsd = feature_store.descriptor feature_store[str(persist_torch_model)] = persist_torch_model.read_bytes()