adding pytorch

qognitive · Aug 27, 2024 · b528d91 · b528d91
1 parent 7f1d41d
commit b528d91
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -39,6 +39,7 @@ dev = [
   "pytest-asyncio",
   "pytest-cov",
   "testcontainers",
+  "torch",
 ]
 examples = [
   "scikit-learn",