fix: dependency

MLRichter · Jan 10, 2022 · 046903f · 046903f
1 parent 3dec91f
commit 046903f
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -33,8 +33,8 @@ numpy = "^1.21.4"
 attrs = "^21.2.0"
 tqdm = "^4.62.3"
 graphviz = "^0.19.1"
-torch = "^1.10.1"
-torchvision = "^0.11.2"
+torch = "*"
+torchvision = "*"
 
 [tool.poetry.extras]
 docs = [
@@ -51,8 +51,8 @@ pytest = "^6.1.2"
 pytest-cov = "^3.0.0"
 pyupgrade = "^2.29.1"
 tox = "^3.20.1"
-torch = "^1.10.1"
-torchvision = "^0.11.2"
+torch = "*"
+torchvision = "*"
 
 [tool.semantic_release]
 branch = "main"