Skip to content

Commit e7fe9a2

Browse files
committed
add missing call to validate data
1 parent 7950f99 commit e7fe9a2

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

pymc_extras/model_builder.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -561,6 +561,8 @@ def predict(
561561
>>> prediction_data = pd.DataFrame({'input':x_pred})
562562
>>> pred_mean = model.predict(prediction_data)
563563
"""
564+
565+
X_pred = self._validate_data(X_pred)
564566

565567
posterior_predictive_samples = self.sample_posterior_predictive(
566568
X_pred, extend_idata, combined=False, predictions=predictions, **kwargs

0 commit comments

Comments
 (0)