Merge pull request #232 from CITCOM-project/test-adequacy

Fixed serialisation bug
CITCOM-project · Aug 15, 2023 · ee8be4d · ee8be4d
2 parents 13a44fc + b1b30be
commit ee8be4d
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/causal_testing/testing/causal_test_result.py b/causal_testing/testing/causal_test_result.py
@@ -85,9 +85,11 @@ def to_dict(self, json=False):
             "outcome": self.estimator.outcome,
             "adjustment_set": list(self.adjustment_set) if json else self.adjustment_set,
             "effect_measure": self.test_value.type,
-            "effect_estimate": self.test_value.value,
-            "ci_low": self.ci_low(),
-            "ci_high": self.ci_high(),
+            "effect_estimate": self.test_value.value.to_dict()
+            if json and hasattr(self.test_value.value, "to_dict")
+            else self.test_value.value,
+            "ci_low": self.ci_low().to_dict() if json and hasattr(self.ci_low(), "to_dict") else self.ci_low(),
+            "ci_high": self.ci_high().to_dict() if json and hasattr(self.ci_high(), "to_dict") else self.ci_high(),
         }
         if self.adequacy:
             base_dict["adequacy"] = self.adequacy.to_dict()