hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 18:41:40 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

Tue Feb 6 20:57:42 PST 2018 (#1521)

Browse source 
The test failure was unrelated
This commit is contained in:
Eric Liang 2018-02-06 23:11:31 -08:00 • committed by Philipp Moritz
parent 4ec51a4660
commit 0a9dbc84b5
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
python/ray/rllib/ppo/rollout.py
View file

@ -25,8 +25,8 @@ def collect_samples(agents, config, local_evaluator):
[fut_sample], _ = ray.wait(list(agent_dict)) [fut_sample], _ = ray.wait(list(agent_dict))
agent = agent_dict.pop(fut_sample) agent = agent_dict.pop(fut_sample)
# Start task with next trajectory and record it in the dictionary. # Start task with next trajectory and record it in the dictionary.
fut_sample = agent.sample.remote() fut_sample2 = agent.sample.remote()
agent_dict[fut_sample] = agent agent_dict[fut_sample2] = agent
next_sample = ray.get(fut_sample) next_sample = ray.get(fut_sample)
num_timesteps_so_far += next_sample.count num_timesteps_so_far += next_sample.count