Skip to content

Commit 2724d94

Browse files
Handle series values
1 parent 767e5b4 commit 2724d94

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

causal_testing/testing/causal_test_outcome.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
from abc import ABC, abstractmethod
66
from collections.abc import Iterable
77
import numpy as np
8+
import pandas as pd
89

910
from causal_testing.testing.causal_test_result import CausalTestResult
1011

@@ -57,7 +58,7 @@ def apply(self, res: CausalTestResult) -> bool:
5758
ci_low = res.ci_low() if isinstance(res.ci_low(), Iterable) else [res.ci_low()]
5859
ci_high = res.ci_high() if isinstance(res.ci_high(), Iterable) else [res.ci_high()]
5960
value = res.test_value.value if isinstance(res.ci_high(), Iterable) else [res.test_value.value]
60-
61+
value = value[0] if isinstance(value[0], pd.Series) else value
6162
return (
6263
sum(
6364
not ((ci_low < 0 < ci_high) or abs(v) < self.atol)

0 commit comments

Comments
 (0)