put entire diconv into threadlimit

cbg-ethz · Oct 15, 2024 · 8e71f09 · 8e71f09
1 parent aa35947
commit 8e71f09
Showing 1 changed file with 10 additions and 10 deletions.
diff --git a/lollipop/cli/deconvolute.py b/lollipop/cli/deconvolute.py
@@ -238,17 +238,17 @@ def _deconvolute_bootstrap(
                 # just run one on everything
                 weights = {}
 
-            # deconvolution
-            t_kdec = ll.KernelDeconv(
-                temp_df2[var_dates["var_dates"][mindate] + ["undetermined"]],
-                temp_df2["frac"],
-                temp_df2["date"],
-                kernel=kernel(**kernel_params),
-                reg=regressor(**regressor_params),
-                confint=confint(**confint_params),
-                **weights,
-            )
             with controller.limit(limits=1, user_api='blas'):
+                # deconvolution
+                t_kdec = ll.KernelDeconv(
+                    temp_df2[var_dates["var_dates"][mindate] + ["undetermined"]],
+                    temp_df2["frac"],
+                    temp_df2["date"],
+                    kernel=kernel(**kernel_params),
+                    reg=regressor(**regressor_params),
+                    confint=confint(**confint_params),
+                    **weights,
+                )
                 t_kdec = t_kdec.deconv_all(**deconv_params)
 
             if have_confint: