Skip to content

Commit 301b28a

Browse files
ricardoV94michaelosthege
authored andcommitted
Fix flaky test_truncation_specialized_op
Using empty for the observed would sometimes lead PyMC into triggering automatic imputation, whenever the trash values in empty were nan.
1 parent 3fcf675 commit 301b28a

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

pymc/tests/distributions/test_truncated.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -84,9 +84,7 @@ def test_truncation_specialized_op(shape_info):
8484
dist=x,
8585
lower=5,
8686
upper=15,
87-
observed=np.empty(
88-
100,
89-
),
87+
observed=np.zeros(100),
9088
)
9189
else:
9290
raise ValueError(f"Not a valid shape_info parametrization: {shape_info}")

0 commit comments

Comments
 (0)