fix

gwastro · Jul 29, 2024 · b4a47af · b4a47af
1 parent b06d32e
commit b4a47af
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/pycbc/inference/models/hierarchical.py b/pycbc/inference/models/hierarchical.py
@@ -759,6 +759,9 @@ def total_loglr(self):
                     sh_others[i] += sh_other
                     hh_others[i] += hh_other
                     other_model.return_sh_hh = False
+                    # set logr, otherwise it will store (sh, hh)
+                    setattr(other_model._current_stats, 'loglr',
+                            other_model.marginalize_loglr(sh_other, hh_other))
             else:
                 # use one margin point set to approximate all the others
                 current_params_other.update(
@@ -769,6 +772,9 @@ def total_loglr(self):
                 other_model.return_sh_hh = True
                 sh_other, hh_other = other_model.loglr
                 other_model.return_sh_hh = False
+                # set logr, otherwise it will store (sh, hh)
+                setattr(other_model._current_stats, 'loglr',
+                        other_model.marginalize_loglr(sh_other, hh_other))
                 sh_others += sh_other
                 hh_others += hh_other