diff --git a/src/mrtool/cov_selection/covfinder.py b/src/mrtool/cov_selection/covfinder.py index 840f343..15aa2bb 100644 --- a/src/mrtool/cov_selection/covfinder.py +++ b/src/mrtool/cov_selection/covfinder.py @@ -179,6 +179,8 @@ def create_model( model = MRBRT( self.data, cov_models=cov_models, inlier_pct=self.inlier_pct ) + model.attach_data() + model._infer_shape() return model def fit_gaussian_model(self, covs: list[str]) -> MRBRT: