bug fix with string name

Signed-off-by: rahulbshrestha <rahulshrestha0101@gmail.com>
py-why · May 14, 2024 · 4c635cd · 4c635cd
1 parent dc1f6c2
commit 4c635cd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dowhy/causal_estimators/propensity_score_matching_estimator.py b/dowhy/causal_estimators/propensity_score_matching_estimator.py
@@ -154,7 +154,7 @@ def estimate_effect(
         atc = 0
         numcontrolunits = control.shape[0]
 
-        outcome_variable = self._target_estimand.outcome_variable[0].item()
+        outcome_variable = self._target_estimand.outcome_variable[0]
         control_outcome = control[outcome_variable]
         treated_outcome = [treated.iloc[i][outcome_variable].item() for i in indices]  # TODO: Further vectorize this