Fix py311

sdv-dev · Jul 6, 2023 · fb2dfff · fb2dfff
1 parent 1f5fca8
commit fb2dfff
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 23 deletions.
diff --git a/.github/workflows/integration.yml b/.github/workflows/integration.yml
@@ -10,19 +10,14 @@ jobs:
     runs-on: ${{ matrix.os }}
     strategy:
       matrix:
-        python-version: ['3.8' '3.9', '3.10', '3.11']
+        python-version: ['3.8', '3.9', '3.10', '3.11']
         os: [ubuntu-latest, macos-latest, windows-latest]
     steps:
     - uses: actions/checkout@v1
     - name: Set up Python ${{ matrix.python-version }}
       uses: actions/setup-python@v2
       with:
         python-version: ${{ matrix.python-version }}
-    - if: matrix.os == 'windows-latest'
-      name: Install dependencies - Windows
-      run: |
-          python -m pip install --upgrade pip
-          python -m pip install 'torch>=1.8,<2' -f https://download.pytorch.org/whl/cpu/torch/
     - name: Install dependencies
       run: |
           python -m pip install --upgrade pip

diff --git a/.github/workflows/minimum.yml b/.github/workflows/minimum.yml
@@ -18,16 +18,6 @@ jobs:
       uses: actions/setup-python@v2
       with:
         python-version: ${{ matrix.python-version }}
-    - if: matrix.os == 'windows-latest' && matrix.python-version < 3.10
-      name: Install dependencies - Windows
-      run: |
-          python -m pip install --upgrade pip
-          python -m pip install 'torch==1.8' -f https://download.pytorch.org/whl/cpu/torch/
-    - if: matrix.os == 'windows-latest' && matrix.python-version >= 3.10
-      name: Install dependencies - Windows
-      run: |
-          python -m pip install --upgrade pip
-          python -m pip install 'torch==1.11.0' -f https://download.pytorch.org/whl/cpu/torch/
     - name: Install dependencies
       run: |
           python -m pip install --upgrade pip

diff --git a/.github/workflows/unit.yml b/.github/workflows/unit.yml
@@ -18,17 +18,12 @@ jobs:
       uses: actions/setup-python@v2
       with:
         python-version: ${{ matrix.python-version }}
-    - if: matrix.os == 'windows-latest'
-      name: Install dependencies - Windows
-      run: |
-          python -m pip install --upgrade pip
-          python -m pip install 'torch>=1.8,<2' -f https://download.pytorch.org/whl/cpu/torch/
     - name: Install dependencies
       run: |
           python -m pip install --upgrade pip
           python -m pip install invoke .[test]
     - name: Run unit tests
       run: invoke unit
-    - if: matrix.os == 'ubuntu-latest' && matrix.python-version == 3.8
+    - if: matrix.os == 'ubuntu-latest' && matrix.python-version == 3.9
       name: Upload codecov report
       uses: codecov/codecov-action@v2
diff --git a/setup.py b/setup.py
@@ -18,7 +18,8 @@
     "pandas>=1.3.4;python_version>='3.10'",
     "scikit-learn>=1.1.3,<2;python_version>='3.10'",
     "torch>=1.8.0;python_version<'3.10'",
-    "torch>=1.11.0;python_version>='3.10'",
+    "torch>=1.11.0;python_version=='3.10'",
+    "torch>=2.0.0;python_version>'3.10'",
     'rdt>=1.3.0,<2.0',
 ]