Skip to content

Commit db49aa7

Browse files
committed
set is_fitted = True after .fit call
1 parent 04b838a commit db49aa7

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
@@ -524,6 +524,8 @@ def fit(
524524
)
525525
self.idata.add_groups(fit_data=combined_data.to_xarray()) # type: ignore
526526

527+
self.is_fitted_ = True
528+
527529
return self.idata # type: ignore
528530

529531
def predict(

0 commit comments

Comments
 (0)