set inner_max_num_threads=1

Author: selmanozleyen

src/squidpy/_utils.py

@@ -187,21 +187,21 @@ def wrapper(*args: Any, **kwargs: Any) -> Any:
                 thread.start()
             else:
                 pbar, queue, thread = None, None, None
-
-            res = jl.Parallel(n_jobs=n_jobs, backend=backend)(
-                jl.delayed(_callback_wrapper)(
-                    *((chosen_runner, i, cs) if use_ixs else (chosen_runner, cs)),
-                    *args,
-                    **kwargs,
-                    queue=queue,
+            with jl.parallel_backend(backend=backend, inner_max_num_threads=1):
+                res = jl.Parallel(n_jobs=n_jobs)(
+                    jl.delayed(_callback_wrapper)(
+                        *((chosen_runner, i, cs) if use_ixs else (chosen_runner, cs)),
+                        *args,
+                        **kwargs,
+                        queue=queue,
+                    )
+                    for i, cs in enumerate(collections)
                 )
-                for i, cs in enumerate(collections)
-            )
 
-            if thread is not None:
-                thread.join()
+                if thread is not None:
+                    thread.join()
 
-            return res if extractor is None else extractor(res)
+                return res if extractor is None else extractor(res)
         finally:
             numba.set_num_threads(original_num_threads)