Skip to content

Commit 3bcefc7

Browse files
Remove gen expression so elements can be indexed
1 parent 8f9499d commit 3bcefc7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

causal_testing/testing/estimators.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -384,7 +384,7 @@ def estimate_ate(self) -> tuple[pd.Series, list[pd.Series, pd.Series]]:
384384
t_test_results = model.t_test(individuals.loc["treated"] - individuals.loc["control"])
385385
ate = pd.Series(t_test_results.effect[0])
386386
confidence_intervals = list(t_test_results.conf_int(alpha=self.alpha).flatten())
387-
confidence_intervals = (pd.Series(interval) for interval in confidence_intervals)
387+
confidence_intervals = [pd.Series(interval) for interval in confidence_intervals]
388388
return ate, confidence_intervals
389389

390390
def estimate_control_treatment(self, adjustment_config: dict = None) -> tuple[pd.Series, pd.Series]:

0 commit comments

Comments
 (0)