maint: add install dependencies for sbi (#4)

samadpls · Nov 26, 2024 · fa2765f · fa2765f
1 parent 81d7e34
commit fa2765f
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/setup.py b/setup.py
@@ -75,18 +75,19 @@ def run(self):
 
 if __name__ == "__main__":
     extras = {
-        'opt': ['scikit-learn'],
-        'parallel': ['joblib', 'psutil'],
-        'test': ['flake8', 'pytest', 'pytest-cov', ],
         'docs': ['mne', 'nibabel', 'pooch', 'tdqm',
                  'sphinx', 'sphinx-gallery',
                  'sphinx_bootstrap_theme', 'sphinx-copybutton',
                  'pillow', 'numpydoc',
                  ],
         'gui': ['ipywidgets>=8.0.0', 'ipykernel', 'ipympl', 'voila', ],
+        'opt': ['scikit-learn'],
+        'parallel': ['joblib', 'psutil'],
+        'sbi': ['sbi'],
+        'test': ['flake8', 'pytest', 'pytest-cov', ],
     }
-    extras['dev'] = (extras['opt'] + extras['parallel'] + extras['test'] +
-                     extras['docs'] + extras['gui']
+    extras['dev'] = (extras['docs'] + extras['gui'] + extras['opt']
+                     + extras['parallel'] + extras['sbi'] + extras['test']
                      )