未验证 提交 5b31b843 编写于 作者: 小桐桐 提交者: GitHub

Fix pcl_rl bugs (#10465)

上级 f08513d4
......@@ -204,8 +204,8 @@ class Controller(object):
observations, actions, rewards,
pads) = self._sample_episodes(sess, greedy=greedy)
observations = zip(*observations)
actions = zip(*actions)
observations = list(zip(*observations))
actions = list(zip(*actions))
terminated = np.array(self.env.dones)
......
......@@ -175,7 +175,7 @@ class EnvSpec(object):
if batched:
return obs
else:
return zip(*obs)[0]
return list(zip(*obs))[0]
def initial_act(self, batch_size=None):
batched = batch_size is not None
......@@ -191,7 +191,7 @@ class EnvSpec(object):
if batched:
return act
else:
return zip(*act)[0]
return list(zip(*act))[0]
def is_discrete(self, typ):
return typ == spaces.discrete
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册