Fix tox.ini for installing test deps on darwin

aws · Oct 15, 2024 · 8412a20 · 8412a20
1 parent 8e4efdb
commit 8412a20
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/tox.ini b/tox.ini
@@ -83,12 +83,15 @@ passenv =
 commands =
     python -c "import os; os.system('install-custom-pkgs --install-boto-wheels')"
     pip install 'apache-airflow==2.9.3' --constraint "https://raw.githubusercontent.com/apache/airflow/constraints-2.9.3/constraints-3.8.txt"
-    pip install 'torch==2.0.1+cpu' -f 'https://download.pytorch.org/whl/torch_stable.html'
-    pip install 'torchvision==0.15.2+cpu' -f 'https://download.pytorch.org/whl/torch_stable.html'
     pip install 'dill>=0.3.8'
-
     pytest {posargs}
-deps = .[test]
+deps = 
+    .[test]
+    torch==2.0.1 ; sys_platform == 'darwin'
+    torchvision==0.15.2 ; sys_platform == 'darwin'
+    torch==2.0.1+cpu ; sys_platform != 'darwin'
+    torchvision==0.15.2+cpu ; sys_platform != 'darwin'
+    --find-links 'https://download.pytorch.org/whl/torch_stable.html'
 depends =
     {py38,py39,py310,p311}: clean