Skip to content
Snippets Groups Projects

Add reflective boundary conditions

Merged Colm Talbot requested to merge reflecting-boundaries into master
2 files
+ 5
3
Compare changes
  • Side-by-side
  • Inline
Files
2
@@ -72,7 +72,8 @@ class JumpProposal(object):
return self._apply_boundaries(sample)
def _move_reflecting_keys(self, sample):
keys = [key for key in sample.keys() if not self.priors[key].boundary]
keys = [key for key in sample.keys()
if self.priors[key].boundary == 'reflective']
for key in keys:
if sample[key] > self.priors[key].maximum or sample[key] < self.priors[key].minimum:
r = self.priors[key].maximum - self.priors[key].minimum
@@ -84,7 +85,8 @@ class JumpProposal(object):
return sample
def _move_periodic_keys(self, sample):
keys = [key for key in sample.keys() if self.priors[key].boundary]
keys = [key for key in sample.keys()
if self.priors[key].boundary == 'periodic']
for key in keys:
if sample[key] > self.priors[key].maximum or sample[key] < self.priors[key].minimum:
sample[key] = (self.priors[key].minimum +
Loading