Merge pull request #286 from STOmics/multiAnalysis

Change the core variable name to num_threads
STOmics · May 28, 2024 · 85339d1 · 85339d1
2 parents d2da126 + 29329e5
commit 85339d1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/stereo/algorithm/total_vi.py b/stereo/algorithm/total_vi.py
@@ -48,7 +48,7 @@ def main(
             rna_use_raw: bool = False,
             protein_use_raw: bool = False,
             use_gpu: Union[int, str, bool] = None,
-            core: int = None,
+            num_threads: int = None,
             train_kwargs: Optional[dict] = {},
             **kwags
     ):
@@ -135,7 +135,7 @@ def main(
             })
 
         total_vi = scvi.model.TOTALVI(mdata, **kwags)
-        scvi.settings.num_threads = core
+        scvi.settings.num_threads = num_threads
         total_vi.train(use_gpu=use_gpu, **train_kwargs)
 
         if not self._use_hvg: