From 028bf22101b1e5cd65643e67dd749a130d34625d Mon Sep 17 00:00:00 2001 From: happyqiu Date: Fri, 16 Jun 2023 16:20:09 -0400 Subject: [PATCH] added n_processes kwarg to caiman.run() --- mesmerize_core/caiman_extensions/common.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/mesmerize_core/caiman_extensions/common.py b/mesmerize_core/caiman_extensions/common.py index 490affe..700a27a 100644 --- a/mesmerize_core/caiman_extensions/common.py +++ b/mesmerize_core/caiman_extensions/common.py @@ -441,6 +441,7 @@ def run( self, backend: Optional[str] = None, wait: bool = True, + n_processes: int = None, **kwargs ): """ @@ -479,6 +480,11 @@ def run( f"{COMPUTE_BACKENDS}" ) + if n_processes is not None: + if not isinstance(n_processes, int): + raise TypeError("n_processes argument must be int") + os.environ["MESMERIZE_N_PROCESSES"] = str(n_processes) + batch_path = self._series.paths.get_batch_path() if backend == COMPUTE_BACKEND_LOCAL: