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

Initializing default saver inside the function (#6540)

Browse source
This commit is contained in:
Eyal Sela 2019-12-19 12:29:45 -08:00 • committed by Simon Mo
parent 634b437543
commit 7b955881f3
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
rllib/rollout.py
View file

@ -297,11 +297,14 @@ def rollout(agent,
env_name,
num_steps,
num_episodes=0,
saver=RolloutSaver(),
saver=None,
no_render=True,
monitor=False):
policy_agent_mapping = default_policy_agent_mapping
if saver is None:
saver = RolloutSaver()
if hasattr(agent, "workers"):
env = agent.workers.local_worker().env
multiagent = isinstance(env, MultiAgentEnv)