Skip to content

Commit d54d28a

Browse files
committed
fix pre-commit
1 parent 76eb637 commit d54d28a

File tree

2 files changed

+6
-7
lines changed

2 files changed

+6
-7
lines changed

causalpy/experiments/base.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
"""
1717

1818
from abc import abstractmethod
19-
from typing import Any, Union
19+
from typing import Any, Literal, Union
2020

2121
import arviz as az
2222
import 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.

docs/source/_static/interrogate_badge.svg

Lines changed: 3 additions & 3 deletions
Loading

0 commit comments

Comments
 (0)