From 23aaf83d79f9ca681e74b5a40219a3e18a403b96 Mon Sep 17 00:00:00 2001 From: gyunt Date: Fri, 22 Mar 2019 05:28:44 +0900 Subject: [PATCH] fix a case when `states` is None. --- baselines/ppo2/model.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/baselines/ppo2/model.py b/baselines/ppo2/model.py index 3ba3e82..bbaa4d8 100644 --- a/baselines/ppo2/model.py +++ b/baselines/ppo2/model.py @@ -136,7 +136,8 @@ class Model(object): def step(self, observations, **kwargs): kwargs.update({'observations': observations}) transition = self.act_model.step(**kwargs) - return transition['actions'], transition['values'], transition['states'], transition['neglogpacs'] + states = transition['states'] if 'states' in transition else None + return transition['actions'], transition['values'], states, transition['neglogpacs'] def train(self, lr,