is_fitted = True when trained model is loaded

butterman0 · butterman0 · commit edec5395359e · 2025-05-24T17:35:59.000+02:00
diff --git a/pymc_extras/model_builder.py b/pymc_extras/model_builder.py
@@ -444,6 +444,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]

Original file line number	Diff line number	Diff line change
`@@ -444,6 +444,7 @@ def load(cls, fname: str):`
`444`	`444`	`sampler_config=json.loads(idata.attrs["sampler_config"]),`
`445`	`445`	`)`
`446`	`446`	`model.idata = idata`
	`447`	`+ model.is_fitted_ = True`
`447`	`448`	`dataset = idata.fit_data.to_dataframe()`
`448`	`449`	`X = dataset.drop(columns=[model.output_var])`
`449`	`450`	`y = dataset[model.output_var]`