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

[RLlib] Fix ignoring "sample_collector" config key (#17460)

Browse source
This commit is contained in:
Kai Fricke 2021-08-04 15:27:35 +01:00 • committed by GitHub
parent 208d997414
commit 5d56a8aac5
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
rllib/evaluation/rollout_worker.py
View file

@ -687,8 +687,7 @@ class RolloutWorker(ParallelIteratorWorker):
soft_horizon=soft_horizon,
no_done_at_end=no_done_at_end,
observation_fn=observation_fn,
sample_collector_class=policy_config.get(
"sample_collector_class"),
sample_collector_class=policy_config.get("sample_collector"),
render=render,
)
# Start the Sampler thread.
@ -708,8 +707,7 @@ class RolloutWorker(ParallelIteratorWorker):
soft_horizon=soft_horizon,
no_done_at_end=no_done_at_end,
observation_fn=observation_fn,
sample_collector_class=policy_config.get(
"sample_collector_class"),
sample_collector_class=policy_config.get("sample_collector"),
render=render,
)