garage.sampler.worker module

Worker interface used in all Samplers.

class Worker(*, seed, max_path_length, worker_number)[source]

Bases: abc.ABC

Worker class used in all Samplers.

collect_rollout()[source]

Collect the current rollout, clearing the internal buffer.

Returns:
Batch of sampled trajectories. May be
truncated if the rollouts haven’t completed yet.
Return type:garage.TrajectoryBatch
rollout()[source]

Sample a single rollout of the agent in the environment.

Returns:
Batch of sampled trajectories. May be
truncated if max_path_length is set.
Return type:garage.TrajectoryBatch
shutdown()[source]

Shutdown the worker.

start_rollout()[source]

Begin a new rollout.

step_rollout()[source]

Take a single time-step in the current rollout.

Returns:True iff the path is done, either due to the environment indicating termination of due to reaching max_path_length.
update_agent(agent_update)[source]

Update the worker’s agent, using agent_update.

Parameters:agent_update (object) – An agent update. The exact type of this argument depends on the Worker implementation.
update_env(env_update)[source]

Update the worker’s env, using env_update.

Parameters:env_update (object) – An environment update. The exact type of this argument depends on the Worker implementation.