Fix tox.ini for installing test deps on darwin

Author: benieric

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