feat: moved blob models inside the repo for test robustness

deel-ai · Apr 23, 2024 · c764ff1 · c764ff1
1 parent 86e7df0
commit c764ff1
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 2 deletions.
diff --git a/assets/pytest_data/.gitkeep b/assets/pytest_data/.gitkeep
diff --git a/assets/pytest_models/blobs_mlp.h5 b/assets/pytest_models/blobs_mlp.h5
diff --git a/assets/pytest_models/blobs_mlp.pt b/assets/pytest_models/blobs_mlp.pt
diff --git a/tests/tests_torch/torch_methods_utils.py b/tests/tests_torch/torch_methods_utils.py
@@ -33,8 +33,8 @@
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-model_path = os.path.expanduser("~/") + ".oodeel/saved_models"
-data_path = os.path.expanduser("~/") + ".oodeel/datasets"
+model_path = "./assets/pytest_models"
+data_path = "./assets/pytest_data"
 os.makedirs(model_path, exist_ok=True)
 os.makedirs(data_path, exist_ok=True)