Skip to content

Commit 9e6ede0

Browse files
committed
update typing
Signed-off-by: Nathaniel <[email protected]>
1 parent 78ed0ce commit 9e6ede0

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

causalpy/variable_selection_priors.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -349,7 +349,7 @@ def create_prior(
349349
dims: str | tuple | None = None,
350350
X: np.ndarray | None = None,
351351
hyperparams: dict[str, Any] | None = None,
352-
) -> pm.Deterministic | pm.Distribution:
352+
) -> pm.Deterministic:
353353
"""
354354
Create the specified prior on a coefficient vector.
355355
@@ -562,7 +562,7 @@ def create_variable_selection_prior(
562562
dims: str | tuple | None = None,
563563
X: np.ndarray | None = None,
564564
hyperparams: dict[str, Any] | None = None,
565-
) -> pm.Deterministic | pm.Distribution:
565+
) -> pm.Deterministic:
566566
"""
567567
Convenience function to create a variable selection prior in one call.
568568

0 commit comments

Comments
 (0)