updating for torch poetry issue

v7labs · Nov 14, 2023 · c3cfa50 · c3cfa50
1 parent 0654ad5
commit c3cfa50
Show file tree

Hide file tree

Showing 2 changed files with 50 additions and 91 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -112,9 +112,6 @@ darwin = "darwin.cli:main"
 "__init__.py" = ["E402", "F401"]
 "**/{tests,docs,tools}/*" = ["E402", "F403"]
 
-[tool.poetry.dependencies.dataclasses]
-markers = "python_version < \"3.7\""
-version = "^0.8"
 
 [tool.poetry.dependencies.nibabel]
 version = "^5.0.0"
@@ -136,11 +133,11 @@ python = ">=3.8"
 optional = true
 
 [tool.poetry.dependencies.torch]
-version = "^2.1.0"
+version = "^2.0.0"
 optional = true
 
 [tool.poetry.dependencies.torchvision]
-version = "^0.16.0"
+version = "^0.15.0"
 optional = true
 
 [tool.poetry.dependencies.black]