(self, physics, random_state)
| 155 | extra_hook(physics, random_state) |
| 156 | |
| 157 | def after_step(self, physics, random_state): |
| 158 | self._task.after_step(physics, random_state) |
| 159 | for entity_hook in self._after_step.entity_hooks: |
| 160 | entity_hook(physics, random_state) |
| 161 | for extra_hook in self._after_step.extra_hooks: |
| 162 | extra_hook(physics, random_state) |
| 163 | |
| 164 | |
| 165 | class _CommonEnvironment: |