update type hints

Author: drbenvincent

causalpy/pymc_experiments.py

@@ -1,4 +1,4 @@
-from typing import Optional, Union
+from typing import 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: Optional[float] = 0.001,
+        epsilon: float | None = 0.001,
         **kwargs,
     ):
         super().__init__(model=model, **kwargs)

causalpy/skl_experiments.py

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