Big Update

jameschapman19 · Apr 18, 2024 · 86e38cb · 86e38cb
1 parent a95f199
commit 86e38cb
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -20,9 +20,10 @@ numpyro = {version = "*", optional = true}
 arviz = {version = "*", optional = true}
 funsor = {version = "*", optional = true}
 torch = [
-     {version = "^2.0.1", platform = "darwin"},
-     {version = "^2.0.1", platform = "linux", source = "torch"},
-     {version = "^2.0.1", platform = "win32", source = "torch"}, ]
+     {version = "<=2.1.0", platform = "darwin"},
+     {version = "<=2.1.0", platform = "linux", source = "torch"},
+     {version = "<=2.1.0", platform = "win32", source = "torch"},
+]
 lightning = "*"
 
 [[tool.poetry.source]]