Skip to content

GitLab

"git@developer.sourcefind.cn:wangsen/mineru.git" did not exist on "2eaa9ca1d3361c2908aecfbe0d385abce37b74fd"
Unverified Commit 5b31b843 authored by 小桐桐's avatar 小桐桐 Committed by GitHub
Browse files

Fix pcl_rl bugs (#10465)

parent f08513d4
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
research/pcl_rl/controller.py
View file @ 5b31b843
...@@ -204,8 +204,8 @@ class Controller(object): ...@@ -204,8 +204,8 @@ class Controller(object):
observations, actions, rewards, observations, actions, rewards,
pads) = self._sample_episodes(sess, greedy=greedy) pads) = self._sample_episodes(sess, greedy=greedy)
observations = zip(*observations) observations = list(zip(*observations))
actions = zip(*actions) actions = list(zip(*actions))
terminated = np.array(self.env.dones) terminated = np.array(self.env.dones)
......
research/pcl_rl/env_spec.py
View file @ 5b31b843
...@@ -175,7 +175,7 @@ class EnvSpec(object): ...@@ -175,7 +175,7 @@ class EnvSpec(object):
if batched: if batched:
return obs return obs
else: else:
return zip(*obs)[0] return list(zip(*obs))[0]
def initial_act(self, batch_size=None): def initial_act(self, batch_size=None):
batched = batch_size is not None batched = batch_size is not None
...@@ -191,7 +191,7 @@ class EnvSpec(object): ...@@ -191,7 +191,7 @@ class EnvSpec(object):
if batched: if batched:
return act return act
else: else:
return zip(*act)[0] return list(zip(*act))[0]
def is_discrete(self, typ): def is_discrete(self, typ):
return typ == spaces.discrete 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