From c2c17ca558c1580b8be1ebf2204bfc73cca0e523 Mon Sep 17 00:00:00 2001
From: Jacob Silterra <silterra@mit.edu>
Date: Wed, 29 Nov 2023 15:04:35 -0500
Subject: [PATCH] Install different packages depending on platform

Don't install cuda packages on Macs.
---
 setup.cfg | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/setup.cfg b/setup.cfg
index 4360137..cb647a5 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -36,8 +36,10 @@ python_requires = >=3.8
 install_requires =
     importlib-metadata; python_version>="3.8"
     numpy==1.24.1
-    torch==1.11.0+cu113
-    torchvision==0.12.0+cu113
+    torch==1.11.0+cu113; sys_platform != "darwin"
+    torch==1.11.0; sys_platform == "darwin"
+    torchvision==0.12.0+cu113; sys_platform != "darwin"
+    torchvision==0.12.0; sys_platform == "darwin"
     pytorch_lightning==1.5.6
     scikit-learn==1.0.2
     tqdm==4.62.3