fix ndim=1 error in samplerKwargs

dflemin3 · Sep 22, 2020 · bade33b · bade33b
1 parent 9035fa4
commit bade33b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/approxposterior/approx.py b/approxposterior/approx.py
@@ -92,6 +92,7 @@ def __init__(self, theta, y, lnprior, lnlike, priorSample, bounds, gp=None,
             ndim = 1
         else:
             ndim = theta.shape[-1]
+        self.ndim = ndim
 
         # Make sure y, theta are valid floats
         if np.any(~np.isfinite(self.theta)) or np.any(~np.isfinite(self.y)):

diff --git a/approxposterior/mcmcUtils.py b/approxposterior/mcmcUtils.py
@@ -43,13 +43,13 @@ def validateMCMCKwargs(ap, samplerKwargs, mcmcKwargs, verbose=False):
     if samplerKwargs is None:
         samplerKwargs = dict()
 
-        samplerKwargs["ndim"] = ap.theta.shape[-1]
+        samplerKwargs["ndim"] = ap.ndim
         samplerKwargs["nwalkers"] = 20 * samplerKwargs["dim"]
         samplerKwargs["log_prob_fn"] = ap._gpll
     else:
         # If user set ndim, ignore it and align it with theta's dimensionality
         samplerKwargs.pop("ndim", None)
-        samplerKwargs["ndim"] = ap.theta.shape[-1]
+        samplerKwargs["ndim"] = ap.ndim
 
         # Initialize other parameters if they're not provided
         try: