Minor corrections

Author: jmafoster1

causal_testing/specification/capabilities.py

@@ -3,14 +3,15 @@
 treatment sequences that operate over time.
 """
 from causal_testing.specification.variable import Variable
+from typing import Any
 
 
 class Capability:
     """
     Data class to encapsulate temporal interventions.
     """
 
-    def __init__(self, variable: Variable, value: any, start_time: int, end_time: float):
+    def __init__(self, variable: Variable, value: Any, start_time: int, end_time: int):
         self.variable = variable
         self.value = value
         self.start_time = start_time

causal_testing/testing/causal_test_adequacy.py

@@ -12,6 +12,9 @@
 from causal_testing.specification.causal_dag import CausalDAG
 from causal_testing.testing.estimators import Estimator
 from causal_testing.testing.causal_test_case import CausalTestCase
+import logging
+
+logger = logging.getLogger(__name__)
 
 
 class DAGAdequacy:
@@ -104,10 +107,13 @@ def measure_adequacy(self):
             try:
                 results.append(self.test_case.execute_test(estimator, None))
             except LinAlgError:
+                logger.warning("Adequacy LinAlgError")
                 continue
             except ConvergenceError:
+                logger.warning("Adequacy ConvergenceError")
                 continue
-            except ValueError:
+            except ValueError as e:
+                logger.warning(f"Adequacy ValueError: {e}")
                 continue
         outcomes = [self.test_case.expected_causal_effect.apply(c) for c in results]
         results = pd.DataFrame(c.to_dict() for c in results)[["effect_estimate", "ci_low", "ci_high"]]

causal_testing/testing/estimators.py

@@ -697,7 +697,7 @@ def setup_fault_t_do(self, individual: pd.DataFrame):
 
         return pd.DataFrame({"fault_t_do": fault_t_do})
 
-    def setup_fault_time(self, individual, perturbation=-0.001):
+    def setup_fault_time(self, individual: pd.DataFrame, perturbation: float = -0.001):
         """
         Return the time at which the event of interest (i.e. a fault) occurred.
         """

tests/specification_tests/test_capabilities.py

@@ -1,8 +1,4 @@
 import unittest
-from enum import Enum
-import z3
-from scipy.stats import norm, kstest
-
 from causal_testing.specification.capabilities import Capability, TreatmentSequence