needs pytorch

ctr26 · ctr26 · commit 83655b611ba8 · 2023-11-16T17:10:32.000Z
diff --git a/pyproject.toml b/pyproject.toml
@@ -18,6 +18,7 @@ python-dateutil = "^2.8.2"
 annotated-types = "^0.6.0"
 typing-extensions = "^4.8.0"
 lxml = "^4.9.3"
+torch = "^2.1.1"
 
 [tool.poetry.group.dev.dependencies]
 pytest-cov = "^4.1.0"
