Skip to content

Commit 8590dc9

Browse files
Update typing
1 parent 18883d8 commit 8590dc9

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

causal_testing/testing/causal_test_result.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ def __init__(
2727
self,
2828
estimator: Estimator,
2929
test_value: TestValue,
30-
confidence_intervals: [float, float] = None,
30+
confidence_intervals: [pd.Series, pd.Series] = None,
3131
effect_modifier_configuration: {Variable: Any} = None,
3232
adequacy=None,
3333
):
@@ -100,15 +100,15 @@ def ci_low(self):
100100
"""Return the lower bracket of the confidence intervals."""
101101
if self.confidence_intervals:
102102
if isinstance(self.confidence_intervals[0], pd.Series):
103-
return self.confidence_intervals[0][0]
103+
return self.confidence_intervals[0].to_list()
104104
return self.confidence_intervals[0]
105105
return None
106106

107107
def ci_high(self):
108108
"""Return the higher bracket of the confidence intervals."""
109109
if self.confidence_intervals:
110110
if isinstance(self.confidence_intervals[1], pd.Series):
111-
return self.confidence_intervals[1][0]
111+
return self.confidence_intervals[1].to_list()
112112
return self.confidence_intervals[1]
113113
return None
114114

0 commit comments

Comments
 (0)