move torch from extra to dependencies

scikit-learn-contrib · Sep 5, 2024 · e54d1a9 · e54d1a9
1 parent 99c93a8
commit e54d1a9
Show file tree

Hide file tree

Showing 2 changed files with 295 additions and 3 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -53,6 +53,7 @@ twine = "3.7.1"
 wheel = "0.37.1"
 category-encoders = "^2.6.3"
 ipykernel = "^6.29.5"
+torch = "^2.4.0"
 
 [tool.poetry.dev-dependencies]
 matplotlib = "3.6.2"
@@ -78,7 +79,6 @@ sphinx_rtd_theme = "1.0.0"
 [tool.poetry.extras]
 tests = ["typed-ast"]
 docs = ["sphinx-markdown-tables"]
-pytorch = ["torch"]
 
 [tool.poetry.urls]
 "Bug Tracker" = "https://github.com/Quantmetry/qolmat"