rename cuda optional requirements to mamba_cuda (#109)

delphi-suite · Apr 3, 2024 · bfce886 · bfce886
1 parent c961701
commit bfce886
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -27,7 +27,7 @@ dependencies = [
 ]
 
 [project.optional-dependencies]
-cuda = [
+mamba_cuda = [
     "mamba_ssm==1.2.0.post1",
     "causal-conv1d==1.2.0.post2",
 ]

diff --git a/requirements.txt b/requirements.txt
@@ -1,3 +1,3 @@
 # this references dependencies in pyproject.toml (including optional cuda dependencies)
 .
-.[cuda]
+.[mamba_cuda]