set is_fitted = True after .fit call

butterman0 · butterman0 · commit db49aa7b2d59 · 2025-05-24T17:35:59.000+02:00
diff --git a/pymc_extras/model_builder.py b/pymc_extras/model_builder.py
@@ -524,6 +524,8 @@ def fit(
             )
             self.idata.add_groups(fit_data=combined_data.to_xarray())  # type: ignore
 
+        self.is_fitted_ = True
+        
         return self.idata  # type: ignore
 
     def predict(

Original file line number	Diff line number	Diff line change
`@@ -524,6 +524,8 @@ def fit(`
`524`	`524`	`)`
`525`	`525`	`self.idata.add_groups(fit_data=combined_data.to_xarray()) # type: ignore`
`526`	`526`
	`527`	`+ self.is_fitted_ = True`
	`528`	`+`
`527`	`529`	`return self.idata # type: ignore`
`528`	`530`
`529`	`531`	`def predict(`