Minor fixes to modelbuilder

butterman0 · web-flow · commit ed19c342f901 · 2025-10-21T10:36:09.000+02:00
diff --git a/pymc_extras/model_builder.py b/pymc_extras/model_builder.py
@@ -446,6 +446,7 @@ def load(cls, fname: str):
             sampler_config=json.loads(idata.attrs["sampler_config"]),
         )
         model.idata = idata
+        model.is_fitted_ = True
         dataset = idata.fit_data.to_dataframe()
         X = dataset.drop(columns=[model.output_var])
         y = dataset[model.output_var]
@@ -526,6 +527,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
`@@ -446,6 +446,7 @@ def load(cls, fname: str):`
`446`	`446`	`sampler_config=json.loads(idata.attrs["sampler_config"]),`
`447`	`447`	`)`
`448`	`448`	`model.idata = idata`
	`449`	`+ model.is_fitted_ = True`
`449`	`450`	`dataset = idata.fit_data.to_dataframe()`
`450`	`451`	`X = dataset.drop(columns=[model.output_var])`
`451`	`452`	`y = dataset[model.output_var]`
`@@ -526,6 +527,8 @@ def fit(`
`526`	`527`	`)`
`527`	`528`	`self.idata.add_groups(fit_data=combined_data.to_xarray()) # type: ignore`
`528`	`529`
	`530`	`+ self.is_fitted_ = True`
	`531`	`+`
`529`	`532`	`return self.idata # type: ignore`
`530`	`533`
`531`	`534`	`def predict(`