Fix extras for cupy

MobileTeleSystems · Jan 23, 2025 · d19a277 · d19a277
1 parent 3374bd1
commit d19a277
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -84,14 +84,15 @@ pytorch-lightning = {version = ">=1.6.0, <3.0.0", optional = true}
 ipywidgets = {version = ">=7.7,<8.2", optional = true}
 plotly = {version="^5.22.0", optional = true}
 nbformat = {version = ">=4.2.0", optional = true}
-cupy-cuda12x = {version = "^13.3.0", python = "<3.13", extras = ["cupy"], optional = true}
+cupy-cuda12x = {version = "^13.3.0", python = "<3.13", optional = true}
 
 
 [tool.poetry.extras]
 lightfm = ["rectools-lightfm"]
 nmslib = ["nmslib", "nmslib-metabrainz"]
 torch = ["torch", "pytorch-lightning"]
 visuals = ["ipywidgets", "plotly", "nbformat"]
+cupy = ["cupy-cuda12x"]
 all = [
     "rectools-lightfm",
     "nmslib", "nmslib-metabrainz",