Added on_step hook

nengo · Apr 18, 2018 · b778e30 · b778e30
1 parent 3095157
commit b778e30
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/nengo_gui/page.py b/nengo_gui/page.py
@@ -501,6 +501,8 @@ def runner(self):
                         self.sim.run_steps(self.sim.max_steps)
                     else:
                         self.sim.step()
+                        if 'on_step' in self.locals:
+                            self.locals['on_step'](self.sim)
                 except Exception as err:
                     if self.finished:
                         return