18 lines
586 B
Python
18 lines
586 B
Python
|
"""Task config shared by model and data code."""
|
||
|
|
|
||
|
|
from dataclasses import dataclass, field
|
||
|
|
from typing import Any, Dict, Optional
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class TaskConfig:
|
||
|
|
"""Robot task definition shared by model construction and data slicing."""
|
||
|
|
|
||
|
|
dof_config: Dict[str, int] = field(default_factory=dict)
|
||
|
|
agent_pos_config: Dict[str, int] = field(default_factory=dict)
|
||
|
|
ar_dof_config: Optional[Dict[str, int]] = None
|
||
|
|
action_horizon: int = 32
|
||
|
|
action_horizon_flow: int = 32
|
||
|
|
noise_scheduler: Optional[Dict[str, Any]] = None
|
||
|
|
use_state_string_representation: bool = False
|