File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -27,7 +27,7 @@ def __init__(
27
27
self ,
28
28
estimator : Estimator ,
29
29
test_value : TestValue ,
30
- confidence_intervals : [float , float ] = None ,
30
+ confidence_intervals : [pd . Series , pd . Series ] = None ,
31
31
effect_modifier_configuration : {Variable : Any } = None ,
32
32
adequacy = None ,
33
33
):
@@ -100,15 +100,15 @@ def ci_low(self):
100
100
"""Return the lower bracket of the confidence intervals."""
101
101
if self .confidence_intervals :
102
102
if isinstance (self .confidence_intervals [0 ], pd .Series ):
103
- return self .confidence_intervals [0 ][ 0 ]
103
+ return self .confidence_intervals [0 ]. to_list ()
104
104
return self .confidence_intervals [0 ]
105
105
return None
106
106
107
107
def ci_high (self ):
108
108
"""Return the higher bracket of the confidence intervals."""
109
109
if self .confidence_intervals :
110
110
if isinstance (self .confidence_intervals [1 ], pd .Series ):
111
- return self .confidence_intervals [1 ][ 0 ]
111
+ return self .confidence_intervals [1 ]. to_list ()
112
112
return self .confidence_intervals [1 ]
113
113
return None
114
114
You can’t perform that action at this time.
0 commit comments