From 5ed110bfb906d6c0ce27ec13fd5ede593bda3467 Mon Sep 17 00:00:00 2001
From: Nora Gilbertson <148926375+n-gilbertson@users.noreply.github.com>
Date: Fri, 6 Sep 2024 10:33:39 -0700
Subject: [PATCH] fixed create_model to attach data and infer shape

---
 src/mrtool/cov_selection/covfinder.py | 2 ++
 1 file changed, 2 insertions(+)

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: