update type hints - epsilon is optional

Author: drbenvincent

causalpy/pymc_experiments.py

@@ -1,4 +1,4 @@
-from typing import Union
+from typing import Optional, Union
 
 import arviz as az
 import matplotlib.pyplot as plt
@@ -565,7 +565,7 @@ def __init__(
         treatment_threshold: float,
         model=None,
         running_variable_name: str = "x",
-        epsilon: float = 0.001,
+        epsilon: Optional[float] = 0.001,
         **kwargs,
     ):
         super().__init__(model=model, **kwargs)

causalpy/skl_experiments.py

@@ -1,3 +1,5 @@
+from typing import Optional
+
 import matplotlib.pyplot as plt
 import numpy as np
 import pandas as pd
@@ -370,7 +372,7 @@ def __init__(
         treatment_threshold,
         model=None,
         running_variable_name="x",
-        epsilon: float = 0.001,
+        epsilon: Optional[float] = 0.001,
         **kwargs,
     ):
         super().__init__(model=model, **kwargs)