From 6d6d53e3df7622e72edd090cf4ce695bc1d7af4d Mon Sep 17 00:00:00 2001 From: Tim Sainburg Date: Fri, 6 Mar 2020 14:25:27 -0800 Subject: [PATCH] catch hopper error on shape so shape doesn't die the only problem with this is that error with shape could go completely unnoticed in this case... --- pyoperant/behavior/shape.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/pyoperant/behavior/shape.py b/pyoperant/behavior/shape.py index 0edbc42d..5d1a0f71 100644 --- a/pyoperant/behavior/shape.py +++ b/pyoperant/behavior/shape.py @@ -2,6 +2,7 @@ import datetime as dt from pyoperant import panels from pyoperant import utils +from pyoperant import components class Shaper(object): """ @@ -250,7 +251,11 @@ def temp(): def reward(self, value, next_state): def temp(): self.log.info('%d\t%d\t%s\t%s'%(self.recent_state, self.response_counter, self.last_response, dt.datetime.now().isoformat(' '))) - self.panel.reward(value=value) + try: + self.panel.reward(value=value) + except components.HopperWontComeUpError as err: + self.panel.reset() + return next_state return temp