fix: Turn cuda into a pip extra (from a Poetry group)

2024-02-09 03:07:54 +01:00
parent 2de7a65d02
commit b6752dd5f7
3 changed files with 18 additions and 12 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -12,11 +12,15 @@ readme = "README.md"
 [tool.poetry.dependencies]
 python = "^3.9,<3.13"
 quimb = { version = "^1.6.0", extras = ["tensor"] }
+cupy = { version = "^11.6.0", optional = true }
+cuquantum-python-cu11 = { version = "^23.3.0", optional = true }
+
+[tool.poetry.extras]
+cuda = ["cupy", "cuquantum-python-cu11"]

 [tool.poetry.group.dev.dependencies]
 ipython = "^7.0.0"

-
 [tool.poetry.group.tests]
 optional = true