Update pyproject.toml

janelia-cellmap · Mar 18, 2024 · 07e06b2 · 07e06b2
1 parent 95ea92a
commit 07e06b2
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -20,8 +20,8 @@ dependencies = [
   'torch',
   'torchvision',
   'numpy',
-  "tqdm",
-  'cellpose[gui]'
+  'tqdm',
+  'cellpose'
 ]
 
 [project.optional-dependencies]
@@ -34,6 +34,8 @@ dev = [
   'pdoc',
   'pre-commit',
 ]
+pretrained = ['cellpose[gui]']
+all = ["cellmap-models[dev,pretrained]"]
 
 [project.urls]
 homepage = "https://janelia-cellmap.github.io/cellmap-models"