Update _em_model.py

Add argument `show_progress_bar` to `fit` method.
theislab · Dec 1, 2023 · d120e50 · d120e50
1 parent b2c8258
commit d120e50
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/scvelo/tools/_em_model.py b/scvelo/tools/_em_model.py
@@ -227,6 +227,7 @@ def fit(
         load_pars: bool = False,
         steady_state_prior: Optional[List[bool]] = None,
         assignment_mode: str = "projection",
+        show_progress_bar: bool = True,
         **kwargs,
     ):
         """Fit the model."""
@@ -259,7 +260,7 @@ def fit(
             unit="gene",
             as_array=False,
             backend=self._backend,
-            show_progress_bar=False,  # len(self._var_names) > 9,
+            show_progress_bar=show_progress_bar,
         )(
             use_raw=use_raw,
             load_pars=load_pars,