added todo comment

Signed-off-by: Rahul Shrestha <rahulshrestha0101@gmail.com>
py-why · May 12, 2024 · 006cfc6 · 006cfc6
1 parent 804bf95
commit 006cfc6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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)