Change hardcoded dataset path to environ variable (#51)

* 🛠 change hardcoded dataset path to environ variable * Add dataset path to environment variable in workflow Co-authored-by: Ashwin Vaidya <ashwinitinvaidya@gmail.com>
openvinotoolkit · Jan 5, 2022 · 2bae059 · 2bae059
1 parent b169c4d
commit 2bae059
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/.github/workflows/tox.yml b/.github/workflows/tox.yml
@@ -22,6 +22,7 @@ jobs:
         run: tox -e black,isort,flake8,pylint,mypy,pydocstyle
       - name: Coverage
         run: |
+          export ANOMALIB_DATASET_PATH=/media/data1/datasets/MVTec
           export CUDA_VISIBLE_DEVICES=3
           tox -e coverage
       - name: Upload coverage result

diff --git a/tests/helpers/dataset.py b/tests/helpers/dataset.py
@@ -23,8 +23,7 @@ def get_dataset_path(path: Union[str, Path] = "./datasets/MVTec"):
     # when running locally
     path = str(path)
     if not os.path.isdir(path):
-        # when using docker image
-        path = "/tmp/anomalib/datasets/MVTec"
+        path = os.environ["ANOMALIB_DATASET_PATH"]
     return path
 
 

diff --git a/tox.ini b/tox.ini
@@ -63,6 +63,7 @@ passenv = ftp_proxy
     HTTP_PROXY
     HTTPS_PROXY
     CUDA_VISIBLE_DEVICES
+    ANOMALIB_DATASET_PATH
 deps =
     coverage
     pytest