Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
WuShichao committed Feb 21, 2024
1 parent 059c37d commit 6ebb2c5
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions pycbc/inference/models/hierarchical.py
Original file line number Diff line number Diff line change
Expand Up @@ -736,7 +736,7 @@ def _loglikelihood(self):
# update parameters in primary_model,
# other_models will be updated in total_loglr,
# because other_models need to handle margin_params
for lbl, _ in self.primary_model.items():
for lbl, _ in enumerate(self.primary_model):
self.primary_model.update(
**{p.subname: self.current_params[p.fullname]
for p in self.param_map[lbl]})
Expand Down Expand Up @@ -929,7 +929,7 @@ def get_loglr():
# update parameters in primary_model,
# other_models will be updated in total_loglr,
# because other_models need to handle margin_params
for lbl, _ in self.primary_model.items():
for lbl, _ in enumerate(self.primary_model):
p = {param.subname: self.current_params[param.fullname]
for param in self.param_map[lbl]}
p.update(rec)
Expand Down

0 comments on commit 6ebb2c5

Please sign in to comment.