fixed create_model to attach data and infer shape

ihmeuw-msca · Sep 6, 2024 · 5ed110b · 5ed110b
1 parent 6afd18c
commit 5ed110b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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: