fix default grade_frac

PolyChord · Jan 10, 2024 · cdaf063 · cdaf063
1 parent a17d78d
commit cdaf063
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/pypolychord/polychord.py b/pypolychord/polychord.py
@@ -553,7 +553,9 @@ def run(loglikelihood, nDims, **kwargs):
  'nlives': {},
  'seed': -1,
  }
- default_kwargs['grade_frac'] = [1.0]*len(default_kwargs['grade_dims'])
+ default_kwargs['grade_frac'] = ([1.0]*len(default_kwargs['grade_dims'])
+ if 'grade_dims' not in kwargs else
+ [1.0]*len(kwargs['grade_dims']))
 
  if not set(kwargs.keys()) <= set(default_kwargs.keys()):
  raise TypeError(f"{__name__} got unknown keyword arguments: "

diff --git a/tests/test_run_pypolychord.py b/tests/test_run_pypolychord.py
@@ -121,11 +121,10 @@ def no_derived_gaussian_likelihood(theta):
 
 def test_grade_dims():
  grade_dims = [1, 3]
- grade_frac = [0.5, 0.5]
  pypolychord.run(gaussian_likelihood, 4, nDerived=1,
  prior=uniform_prior, read_resume=False,
- grade_dims=grade_dims, grade_frac=grade_frac)
+ grade_dims=grade_dims)
  with pytest.raises(ValueError):
  pypolychord.run(gaussian_likelihood, 5, nDerived=1,
  prior=uniform_prior, read_resume=False,
- grade_dims=grade_dims, grade_frac=grade_frac)
+ grade_dims=grade_dims)