bump torch python

clementchadebec · Oct 22, 2023 · 6be08a2 · 6be08a2
1 parent 5e19262
commit 6be08a2
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 10 deletions.
diff --git a/.github/workflows/coverage.yml b/.github/workflows/coverage.yml
@@ -11,13 +11,13 @@ jobs:
         os: [ubuntu-latest]
     env:
       OS: ${{ matrix.os }}
-      PYTHON: '3.7'
+      PYTHON: '3.11'
     steps:
     - uses: actions/checkout@main
     - name: Setup Python
       uses: actions/setup-python@main
       with:
-        python-version: 3.7
+        python-version: 3.11
     - name: Generate coverage report
       run: |
         pip install .

diff --git a/.github/workflows/tests_bench.yml b/.github/workflows/tests_bench.yml
@@ -9,13 +9,13 @@ jobs:
         os: [ubuntu-latest, macos-latest, windows-latest]
     env:
       OS: ${{ matrix.os }}
-      PYTHON: '3.7'
+      PYTHON: '3.11'
     steps:
     - uses: actions/checkout@main
     - name: Setup Python
       uses: actions/setup-python@main
       with:
-        python-version: 3.7
+        python-version: 3.11
     - name: Run tests
       run: |
         pip install .

diff --git a/requirements.txt b/requirements.txt
@@ -4,8 +4,7 @@ numpy>=1.19
 pydantic>=2.0
 scikit-learn
 scipy>=1.7.1
-torch>=1.10.1
+torch>=2.0.1
 tqdm
 typing_extensions
 dataclasses>=0.6
-pickle5; python_version == '3.7.*'
diff --git a/setup.py b/setup.py
@@ -17,9 +17,9 @@
     },
     classifiers=[
         "Programming Language :: Python :: 3",
-        "Programming Language :: Python :: 3.7",
-        "Programming Language :: Python :: 3.8",
         "Programming Language :: Python :: 3.9",
+        "Programming Language :: Python :: 3.10",
+        "Programming Language :: Python :: 3.11",
         "License :: OSI Approved :: Apache Software License",
         "Operating System :: OS Independent",
     ],
@@ -32,11 +32,11 @@
         "pydantic>=2.0",
         "scikit-learn",
         "scipy>=1.7.1",
-        "torch>=1.10.1",
+        "torch>=2.0.1",
         "tqdm",
         "typing_extensions",
         "dataclasses>=0.6",
     ],
     extras_require={':python_version == "3.7.*"': ["pickle5"]},
-    python_requires=">=3.7",
+    python_requires=">=3.9",
 )