needs pytorch

Author: ctr26

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"