protomotions.agents.utils.step_tracker module#

class protomotions.agents.utils.step_tracker.StepTracker(num_envs, min_steps, max_steps, device)[source]#

Bases: <Mock object at 0x7fdd7406d2d0>[]

steps: MockTensor#

__init__(num_envs, min_steps, max_steps, device)[source]#

advance()[source]#

done_indices()[source]#

reset_steps(env_ids=None)[source]#

shift_counter(env_ids, shift)[source]#

property device: <Mock object at 0x7fdd7407f650>[]#: Get device from registered buffers.