diff --git a/tupak/core/sampler.py b/tupak/core/sampler.py index fbba50cba8b31bf199be94146a9f18acc0012fb0..82216f8d32e17868122c97e3ec389576689340a4 100644 --- a/tupak/core/sampler.py +++ b/tupak/core/sampler.py @@ -1201,12 +1201,15 @@ class Pymc3(Sampler): # perform the sampling trace = pymc3.sample(self.draws, step=sm, **self.kwargs) - nparams = len(self.priors.keys()) + nparams = len([key for key in self.priors.keys() if self.priors[key].__class__.__name__ != 'DeltaFunction']) nsamples = len(trace)*self.chains self.result.samples = np.zeros((nsamples, nparams)) - for i, key in enumerate(self.priors.keys()): - self.result.samples[:,i] = trace[key] + count = 0 + for key in self.priors.keys(): + if self.priors[key].__class__.__name__ != 'DeltaFunction': # ignore DeltaFunction variables + self.result.samples[:,count] = trace[key] + count += 1 self.result.sampler_output = np.nan self.calculate_autocorrelation(self.result.samples)