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",
 )