add missing call to validate data

butterman0 · butterman0 · commit 6440dabc3538 · 2025-04-09T16:54:44.000+02:00
diff --git a/pymc_extras/model_builder.py b/pymc_extras/model_builder.py
@@ -558,6 +558,7 @@ def predict(
         >>> pred_mean = model.predict(prediction_data)
         """
 
+        X_pred = self._validate_data(X_pred)
         posterior_predictive_samples = self.sample_posterior_predictive(
             X_pred, extend_idata, combined=False, **kwargs
         )

Original file line number	Diff line number	Diff line change
`@@ -558,6 +558,7 @@ def predict(`
`558`	`558`	`>>> pred_mean = model.predict(prediction_data)`
`559`	`559`	`"""`
`560`	`560`
	`561`	`+ X_pred = self._validate_data(X_pred)`
`561`	`562`	`posterior_predictive_samples = self.sample_posterior_predictive(`
`562`	`563`	`X_pred, extend_idata, combined=False, **kwargs`
`563`	`564`	`)`