From 006cfc614ad5907852495d0b6d3fef3804026431 Mon Sep 17 00:00:00 2001 From: Rahul Shrestha Date: Sun, 12 May 2024 23:48:07 +0200 Subject: [PATCH] added todo comment Signed-off-by: Rahul Shrestha --- .../causal_estimators/propensity_score_matching_estimator.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/dowhy/causal_estimators/propensity_score_matching_estimator.py b/dowhy/causal_estimators/propensity_score_matching_estimator.py index b942a0fcb..762a1431d 100755 --- a/dowhy/causal_estimators/propensity_score_matching_estimator.py +++ b/dowhy/causal_estimators/propensity_score_matching_estimator.py @@ -134,7 +134,7 @@ def estimate_effect( outcome_variable = self._target_estimand.outcome_variable[0] treated_outcomes = treated[outcome_variable] - control_outcomes = [control.iloc[i][outcome_variable].item() for i in indices] + control_outcomes = [control.iloc[i][outcome_variable].item() for i in indices] #TODO: Further vectorize this att = np.mean(treated_outcomes - control_outcomes) @@ -156,7 +156,7 @@ def estimate_effect( outcome_variable = self._target_estimand.outcome_variable[0].item() control_outcome = control[outcome_variable] - treated_outcome = [treated.iloc[i][outcome_variable].item() for i in indices] + treated_outcome = [treated.iloc[i][outcome_variable].item() for i in indices] # TODO: Further vectorize this atc = np.mean(treated_outcome - control_outcome)