"vscode:/vscode.git/clone" did not exist on "6ef71d491fef989cdb82b1fc443208531016e05d"
Unverified Commit 5b31b843 authored by 小桐桐's avatar 小桐桐 Committed by GitHub
Browse files

Fix pcl_rl bugs (#10465)

parent 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% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment