Merge pull request #44 from kenki931128/hotfix/use-correct-variable

use correct variable
kenki931128 · Apr 23, 2024 · 8c3dccb · 8c3dccb
2 parents 85e148d + 915a1a6
commit 8c3dccb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/synthx/method.py b/synthx/method.py
@@ -181,7 +181,7 @@ def process_placebo(
             )
             return None
 
-    results = Parallel(n_jobs=-1)(
+    results = Parallel(n_jobs=n_jobs)(
         delayed(process_placebo)(test_unit_placebo) for test_unit_placebo in tqdm(control_units)
     )
 
@@ -222,7 +222,7 @@ def sensitivity_check(
             )
             .then(pl.col(dataset.y_column) * uplift)
             .otherwise(pl.col(dataset.y_column))
-            .alias('y')
+            .alias(dataset.y_column)
         )
 
         dataset_sensitivity = sx.Dataset(