Merge pull request #240 from CITCOM-project/fix-linalg-error

Fixed linalg error

Author: jmafoster1

causal_testing/testing/causal_test_case.py

@@ -1,6 +1,7 @@
 """This module contains the CausalTestCase class, a class that holds the information required for a causal test"""
 import logging
 from typing import Any
+import numpy as np
 
 from causal_testing.specification.variable import Variable
 from causal_testing.testing.causal_test_outcome import CausalTestOutcome
@@ -79,15 +80,21 @@ def _return_causal_test_results(self, estimator) -> CausalTestResult:
         if not hasattr(estimator, f"estimate_{self.estimate_type}"):
             raise AttributeError(f"{estimator.__class__} has no {self.estimate_type} method.")
         estimate_effect = getattr(estimator, f"estimate_{self.estimate_type}")
-        effect, confidence_intervals = estimate_effect(**self.estimate_params)
-        causal_test_result = CausalTestResult(
-            estimator=estimator,
-            test_value=TestValue(self.estimate_type, effect),
-            effect_modifier_configuration=self.effect_modifier_configuration,
-            confidence_intervals=confidence_intervals,
-        )
-
-        return causal_test_result
+        try:
+            effect, confidence_intervals = estimate_effect(**self.estimate_params)
+            return CausalTestResult(
+                estimator=estimator,
+                test_value=TestValue(self.estimate_type, effect),
+                effect_modifier_configuration=self.effect_modifier_configuration,
+                confidence_intervals=confidence_intervals,
+            )
+        except np.linalg.LinAlgError:
+            return CausalTestResult(
+                estimator=estimator,
+                test_value=TestValue(self.estimate_type, "LinAlgError"),
+                effect_modifier_configuration=self.effect_modifier_configuration,
+                confidence_intervals=None,
+            )
 
     def __str__(self):
         treatment_config = {self.treatment_variable.name: self.treatment_value}