Skip to content

Commit e968e1f

Browse files
Pyrsosmichaelosthege
authored andcommitted
Inferencedata warning (#4493)
* Pass stacklevel argument corrently * Pre-commit errors * Missing comma
1 parent 47b3658 commit e968e1f

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

pymc3/sampling.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -464,9 +464,10 @@ def sample(
464464
v = packaging.version.parse(pm.__version__)
465465
if v.release[0] > 3 or v.release[1] >= 10: # type: ignore
466466
warnings.warn(
467-
"In an upcoming release, pm.sample will return an `arviz.InferenceData` object instead of a `MultiTrace` by default. "
467+
"In v4.0, pm.sample will return an `arviz.InferenceData` object instead of a `MultiTrace` by default. "
468468
"You can pass return_inferencedata=True or return_inferencedata=False to be safe and silence this warning.",
469469
FutureWarning,
470+
stacklevel=2,
470471
)
471472
# set the default
472473
return_inferencedata = False

0 commit comments

Comments
 (0)