ray.rllib.evaluation.rollout_worker.RolloutWorker.set_state#

RolloutWorker.set_state(state: dict) None[source]#

Restores this EnvRunner’s state from the given state dict.

Parameters:

state – The state dict to restore the state from.

from ray.rllib.env.env_runner import EnvRunner
env_runner = ...
state = env_runner.get_state()
new_runner = EnvRunner(...)
new_runner.set_state(state)