Remove torch dependency

pytorch-labs · Nov 21, 2024 · 0dc5d12 · 0dc5d12
1 parent 44f56f1
commit 0dc5d12
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/setup.py b/setup.py
@@ -108,11 +108,11 @@ def make_extension():
 
 
 def get_requirements():
-    pytorch_dep = os.getenv("TORCH_PACKAGE_NAME", "torch")
-    if os.getenv("PYTORCH_VERSION"):
-        pytorch_dep += "==" + os.getenv("PYTORCH_VERSION")
-
-    return [pytorch_dep]
+    if pytorch_dep := os.getenv("TORCH_PACKAGE_NAME"):
+        if os.getenv("PYTORCH_VERSION"):
+            pytorch_dep += "==" + os.getenv("PYTORCH_VERSION")
+            return [pytorch_dep]
+    return []
 
 
 if __name__ == "__main__":