feature(actions): Pytorch Optional.

acederberg · Sep 25, 2024 · a9514e8 · a9514e8
1 parent fe45a81
commit a9514e8
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -26,9 +26,12 @@ pyright = "^1.1.377"
 scikit-learn = "^1.5.1"
 pyyaml = "^6.0.2"
 pyarrow = "^17.0.0"
+pytest = "^8.3.3"
+
+
+[tool.poetry.group.torch]
 torch = {version = "^2.4.1+cpu", source = "pytorch_cpu"}
 torchvision = {version = "^0.19.1+cpu", source = "pytorch_cpu"}
-pytest = "^8.3.3"
 
 
 [tool.poetry.group.ci.dependencies]
@@ -44,6 +47,7 @@ name = "pytorch_cpu"
 url = "https://download.pytorch.org/whl/cpu"
 priority = "explicit"
 
+
 [tool.bumpver]
 current_version = "0.0.3"
 version_pattern = "MAJOR.MINOR.PATCH"