Fix dataset downloads

adaptive-machine-learning · Mar 21, 2024 · 34bdf8e · 34bdf8e
1 parent e07dca8
commit 34bdf8e
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/.github/workflows/unit_test.yml b/.github/workflows/unit_test.yml
@@ -11,6 +11,9 @@ on:
   pull_request:
     branches: [ "main" ]
 
+env:
+  CAPYMOA_DATASETS_DIR: ${{ vars.HOME }}/datasets
+
 jobs:
   tests:
     timeout-minutes: 20
@@ -33,6 +36,12 @@ jobs:
       run: |
         python -m pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu
         python -m pip install ".[dev]"
+    - name: Cache Test Datasets
+      id: cache-test-datasets
+      uses: actions/cache@v4
+      with:
+        path: ${{ vars.CAPYMOA_DATASETS_DIR }}
+        key: "test-datasets"
     - name: PyTest
       run: invoke test.unit
     - name: Notebooks

diff --git a/src/capymoa/stream/PytorchStream.py b/src/capymoa/stream/PytorchStream.py
@@ -30,6 +30,7 @@ class PytorchStream(Stream):
         ...     download=True,
         ...     transform=ToTensor()
         ... )
+        ...
         >>> pytorch_stream = PytorchStream(dataset=pytorchDataset)
         >>> pytorch_stream.get_schema()
         @relation PytorchDataset