From 51a1ef603f35973e20aa7e570e013a263178f299 Mon Sep 17 00:00:00 2001
From: Madhava Jay <me@madhavajay.com>
Date: Thu, 26 Oct 2023 16:35:54 +1000
Subject: [PATCH] Trying cpu extras to speed up pip install on tox tasks

---
 packages/syft/setup.cfg | 2 +-
 tox.ini                 | 5 ++---
 2 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/packages/syft/setup.cfg b/packages/syft/setup.cfg
index 009c7a3a47e..e3a9774ca8e 100644
--- a/packages/syft/setup.cfg
+++ b/packages/syft/setup.cfg
@@ -62,7 +62,7 @@ syft =
     safetensors==0.4.0
     transformers==4.34.0
     evaluate==0.4.1
-    torch==2.1.0
+    torch[cpu]==2.1.0
     recordlinkage==0.16
     argon2-cffi==23.1.0
 
diff --git a/tox.ini b/tox.ini
index 2a8ceeaa162..eead6745fe8 100644
--- a/tox.ini
+++ b/tox.ini
@@ -26,13 +26,12 @@ envlist =
     syftcli.build
 skipsdist = True
 
+
 [testenv]
 basepython = python3
-install_command = pip install --find-links https://download.pytorch.org/whl/cpu/torch_stable.html {opts} {packages}
+install_command = pip install {opts} {packages}
 commands =
     python --version
-setenv =
-    PIP_FIND_LINKS=https://download.pytorch.org/whl/cpu/torch_stable.html
 
 # Syft
 [testenv:syft]