set inner_max_num_threads only for loky

Author: selmanozleyen

src/squidpy/_utils.py

@@ -187,7 +187,8 @@ def wrapper(*args: Any, **kwargs: Any) -> Any:
                 thread.start()
             else:
                 pbar, queue, thread = None, None, None
-            with jl.parallel_backend(backend=backend, inner_max_num_threads=1):
+            kwargs = {} if backend != "loky" else {"inner_max_num_threads": 1}
+            with jl.parallel_backend(backend=backend, **kwargs):
                 res = jl.Parallel(n_jobs=n_jobs)(
                     jl.delayed(_callback_wrapper)(
                         *((chosen_runner, i, cs) if use_ixs else (chosen_runner, cs)),