using ~/.cache/huggingface instead of ./data/cache

angelolab · Feb 17, 2024 · d310679 · d310679
1 parent a7c69fd
commit d310679
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -31,7 +31,7 @@ jobs:
         env:
           cache-name: dataset
         with:
-          path: ./data/cache/
+          path: ~/.cache/huggingface/datasets
           key: huggingface-${{env.cache-name}}
           enableCrossOsArchive: true
           fail-on-cache-miss: true

diff --git a/conftest.py b/conftest.py
@@ -10,7 +10,7 @@
 def dataset_cache_dir() -> Iterator[Union[str, None]]:
     # Change cache directory if running on CI
     if os.environ.get("CI", None):
-        cache_dir = pathlib.Path("./data/cache/")
+        cache_dir = pathlib.Path("~/.cache/huggingface/datasets").expanduser()
     else:
         cache_dir = None
     yield cache_dir

diff --git a/tests/utils/example_dataset_test.py b/tests/utils/example_dataset_test.py
@@ -33,7 +33,7 @@ def dataset_download(request, dataset_cache_dir) -> Iterator[ExampleDataset]:
         Iterator[ExampleDataset]: The iterable Example Dataset.
     """
     # Set up ExampleDataset class
-    example_dataset: ExampleDataset = ExampleDataset(
+    example_dataset = ExampleDataset(
         dataset=request.param,
         cache_dir=dataset_cache_dir,
         revision=EXAMPLE_DATASET_REVISION