Fixed dimensionality for more than one parameter.

CobayaSampler · Apr 15, 2021 · e9ab096 · e9ab096
1 parent e58a292
commit e9ab096
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/cobaya/samplers/polychord/polychord.py b/cobaya/samplers/polychord/polychord.py
@@ -177,8 +177,9 @@ def initialize(self):
         pc_args.pop(pc_args.index("num_repeats"))
 
         if self.use_supernest:
-            self.grade_dims.append(1)
-            self.grade_frac.append(1.0)
+            # For the choice probability, and choice indicator. 
+            self.grade_dims.extend([1, 1])
+            self.grade_frac.extend([1.0, 1.0])
             self.pc_settings = PolyChordSettings(
                 self.nDims + 1,  # FIXME: only true for one proposal.
                 self.nDerived,