Skip to content

Commit

Permalink
Merge pull request #59 from ihmeuw-msca/bugfix/covfinder
Browse files Browse the repository at this point in the history
fixed create_model to attach data and infer shape
  • Loading branch information
zhengp0 authored Oct 7, 2024
2 parents 6afd18c + 5ed110b commit 01fc880
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions src/mrtool/cov_selection/covfinder.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down

0 comments on commit 01fc880

Please sign in to comment.