diff --git a/setup.py b/setup.py index acf4d52915..b5d9c3e505 100644 --- a/setup.py +++ b/setup.py @@ -9,7 +9,7 @@ with open(directory / 'README.md', encoding='utf-8') as f: testing_minimal = [ "numpy", - "torch", + "torch==2.7.1", "pytest", "pytest-xdist", "hypothesis",