Skip to content

Commit 5259105

Browse files
committed
Fix isinstance check in test_predict
Windows returns `np.float32`, which is not an instance of `float`
1 parent 2c30c52 commit 5259105

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pymc_experimental/tests/test_model_builder.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -127,7 +127,7 @@ def test_predict():
127127
assert "y_model" in pred
128128
assert isinstance(pred, dict)
129129
assert len(prediction_data.input.values) == len(pred["y_model"])
130-
assert isinstance(pred["y_model"][0], float)
130+
assert isinstance(pred["y_model"][0], (np.float32, np.float64))
131131

132132

133133
def test_predict_posterior():

0 commit comments

Comments
 (0)