aboutsummaryrefslogtreecommitdiffstats
path: root/simulation/vi_blocks.py
diff options
context:
space:
mode:
Diffstat (limited to 'simulation/vi_blocks.py')
-rw-r--r--simulation/vi_blocks.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/simulation/vi_blocks.py b/simulation/vi_blocks.py
index 3f88611..58b68d3 100644
--- a/simulation/vi_blocks.py
+++ b/simulation/vi_blocks.py
@@ -19,9 +19,9 @@ class ClippedParams(blocks.algorithms.StepRule):
def compute_step(self, parameter, previous_step):
min_clipped = tsr.switch(parameter - previous_step < self.min_value,
- self.min_value, previous_step)
+ 0, previous_step)
return tsr.switch(parameter - previous_step > self.max_value,
- self.max_value, min_clipped)
+ 0, min_clipped), []
def create_vi_model(n_nodes, n_samp=100):