File tree Expand file tree Collapse file tree 2 files changed +6
-7
lines changed
Expand file tree Collapse file tree 2 files changed +6
-7
lines changed Original file line number Diff line number Diff line change 1616"""
1717
1818from abc import abstractmethod
19- from typing import Any , Union
19+ from typing import Any , Literal , Union
2020
2121import arviz as az
2222import matplotlib .pyplot as plt
@@ -54,7 +54,6 @@ class BaseExperiment:
5454 supports_bayes : bool
5555 supports_ols : bool
5656
57-
5857 def __init__ (self , model : Union [PyMCModel , RegressorMixin ] | None = None ) -> None :
5958 # Ensure we've made any provided Scikit Learn model (as identified as being type
6059 # RegressorMixin) compatible with CausalPy by appending our custom methods.
@@ -147,8 +146,8 @@ def effect_summary(
147146 alpha : float = 0.05 ,
148147 cumulative : bool = True ,
149148 relative : bool = True ,
150- min_effect : Optional [ float ] = None ,
151- treated_unit : Optional [ str ] = None ,
149+ min_effect : float | None = None ,
150+ treated_unit : str | None = None ,
152151 ) -> EffectSummary :
153152 """
154153 Generate a decision-ready summary of causal effects.
You can’t perform that action at this time.
0 commit comments