diff --git a/requirements_dev.txt b/requirements_dev.txt index 93aaa1e..03b1b92 100644 --- a/requirements_dev.txt +++ b/requirements_dev.txt @@ -3,6 +3,7 @@ wheel==0.40.0 black==23.3.0 flake8==6.0.0 numpy==1.23.2 +-f https://download.pytorch.org/whl/cu117/torch_stable.html torch==2.0.1 twine==4.0.2 pytest==7.3.2 diff --git a/setup.py b/setup.py index 54987e9..168ec79 100644 --- a/setup.py +++ b/setup.py @@ -4,7 +4,7 @@ readme = readme_file.read() requirements = [ - "numpy>=1.20<1.24", + "numpy>=1.20,<1.24", "torch>=2.0.1", "schnetpack==1.0.1", "ase>=3.22.0",