Skip to content

Commit a012229

Browse files
committed
Upgrade torch version
1 parent da12aaf commit a012229

File tree

2 files changed

+3
-4
lines changed

2 files changed

+3
-4
lines changed

.github/workflows/build.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ jobs:
1818
# We aim to support the versions on pytorch.org
1919
# as well as selected previous versions on
2020
# https://pytorch.org/get-started/previous-versions/
21-
torch-version: ["2.2.2", "2.4.0"]
21+
torch-version: ["2.4.0", "2.6.0"]
2222
sklearn-version: ["latest"]
2323
include:
2424
- os: windows-latest

setup.cfg

+2-3
Original file line numberDiff line numberDiff line change
@@ -31,12 +31,11 @@ where =
3131
python_requires = >=3.9
3232
install_requires =
3333
joblib
34-
numpy<2.0;torch<"2.4"
35-
numpy;torch>="2.4"
34+
numpy
3635
literate-dataclasses
3736
scikit-learn
3837
scipy
39-
torch
38+
torch>=2.4.0
4039
tqdm
4140
matplotlib
4241
requests

0 commit comments

Comments
 (0)