Source code for wbc_mjlab.tasks.config

"""WBC task config: one mjlab task id bound to an env builder."""

from __future__ import annotations

from collections.abc import Callable
from dataclasses import dataclass
from typing import Any

from wbc_mjlab.robots.ids import RobotId

# Avoid importing mjlab here — tasks.config is loaded while ``wbc_mjlab.tasks``
# initializes, and mjlab loads the ``wbc_mjlab.mjlab_entry`` hook (circular import).
EnvCfgBuilder = Callable[[], Any]



[docs]
@dataclass(frozen=True)
class WbcTaskConfig:
  """One registered mjlab task: metadata + env builder."""

  task_id: str
  """CLI / registry id (e.g. ``\"Wbc-G1\"``, ``\"Wbc-H2-Zest\"``)."""
  robot_id: RobotId
  """Robot this task binds to (must be registered)."""
  description: str
  """Short human-readable summary for listings."""
  experiment_name: str
  """Log / experiment directory name."""
  build_env_cfg: EnvCfgBuilder
  """Zero-arg builder returning the env config."""