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 0x7ff4133953d0>[]

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 0x7ff41338d6d0>[]¶: Get device from registered buffers.