Class TrackedEntity

dm_control/composer/hooks_test_utils.py:237–249

A `composer.Entity` that tracks call order of callbacks.

class TrackedEntity(HooksTracker, composer.Entity):
  """A `composer.Entity` that tracks call order of callbacks."""

  def _build(self, name):
    self._mjcf_root = mjcf.RootElement(model=name)

  @property
  def mjcf_model(self):
    return self._mjcf_root

  @property
  def name(self):
    return self._mjcf_root.model


class TrackedTask(HooksTracker, composer.NullTask):