remove pytorch version pin

robsdavis · robsdavis · commit 5294f4eca0d2 · 2025-02-03T10:10:12.000Z
diff --git a/.github/workflows/yank.yml b/.github/workflows/yank.yml
diff --git a/setup.cfg b/setup.cfg
@@ -39,8 +39,8 @@ install_requires =
     scipy
     suod
     tensorflow
-    torch==1.10.0
-    torchvision==0.11.1
+    torch
+    torchvision
     wandb
     xlrd==2.0.1
 
diff --git a/src/data_suite/version.py b/src/data_suite/version.py
@@ -1,4 +1,4 @@
-__version__ = "0.2.0"
+__version__ = "0.2.1"
 
 MAJOR_VERSION = ".".join(__version__.split(".")[:-1])
 PATCH_VERSION = __version__.split(".")[-1]
