Skip to content

Commit ec0597f

Browse files
author
Goose
committed
correct overload type hinting
1 parent b512717 commit ec0597f

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

pymc/sampling/forward.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020
from collections.abc import Callable, Iterable, Sequence
2121
from typing import (
2222
Any,
23+
Literal,
2324
TypeAlias,
2425
cast,
2526
overload,
@@ -367,7 +368,7 @@ def sample_prior_predictive(
367368
model: Model | None = None,
368369
var_names: Iterable[str] | None = None,
369370
random_seed: RandomState = None,
370-
return_inferencedata: bool = True,
371+
return_inferencedata: Literal[True] = True,
371372
idata_kwargs: dict | None = None,
372373
compile_kwargs: dict | None = None,
373374
samples: int | None = None,
@@ -378,7 +379,7 @@ def sample_prior_predictive(
378379
model: Model | None = None,
379380
var_names: Iterable[str] | None = None,
380381
random_seed: RandomState = None,
381-
return_inferencedata: bool = False,
382+
return_inferencedata: Literal[False] = False,
382383
idata_kwargs: dict | None = None,
383384
compile_kwargs: dict | None = None,
384385
samples: int | None = None,

0 commit comments

Comments
 (0)