fix a bug to save model current stats for maxL polarization

gwastro · Jan 31, 2025 · b631e61 · b631e61
1 parent bf5ff50
commit b631e61
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pycbc/inference/models/marginalized_gaussian_noise.py b/pycbc/inference/models/marginalized_gaussian_noise.py
@@ -504,7 +504,7 @@ def _loglr(self):
         # store the maxl polarization
         setattr(self._current_stats,
                 'maxl_polarization',
-                params['polarization'])
+                params['polarization'][idx])
         setattr(self._current_stats, 'maxl_loglr', maxl)
 
         # just store the maxl optimal snrsq