Merge branch 'develop' into add_Aug_DiskCachedDataset

.github/workflows/ci-pipeline.yml

@@ -9,7 +9,7 @@ jobs:
       fail-fast: false
       matrix:
         os: [ubuntu-latest, windows-2022]
-        python-version: ["3.8", "3.9", "3.10", "3.11"]
+        python-version: ["3.8", "3.10", "3.11"]
     steps:
       - uses: actions/checkout@v3
       - if: matrix.os == 'ubuntu-latest'
@@ -21,7 +21,6 @@ jobs:
           python-version: ${{ matrix.python-version }}
       - name: Install requirements
         run: |
-          pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu
           pip install -r test/requirements.txt
           pip install -r test/torch_requirements.txt
           pip install .
@@ -43,7 +42,6 @@ jobs:
           python-version: 3.9
       - name: Generate coverage report
         run: |
-          pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu
           pip install -r test/requirements.txt
           pip install -r test/torch_requirements.txt
           pip install .
@@ -65,8 +63,8 @@ jobs:
           python-version: 3.9
       - name: Install dependencies
         run: |
-          pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu
           pip install -r docs/requirements.txt
+          pip install -r test/torch_requirements.txt
           pip install .
       - name: Build documentation
         run: cd docs && make clean && make html # Use SPHINXOPTS="-W" to fail on warning.

docs/requirements.txt

@@ -4,8 +4,6 @@ sphinx-book-theme
 sphinx-gallery
 myst_nb
 pbr
-torchvision
 ipywidgets
 matplotlib
-torchdata
 sphinx-autoapi

test/torch_requirements.txt

@@ -1,4 +1,5 @@
 --index-url https://download.pytorch.org/whl/cpu
 torch==2.1.0
 torchaudio==2.1.0
+torchvision==0.16.0
 torchdata