Update Wall-X to 1.1.0 (#104)

wall_x/config/task_config.py

@@ -0,0 +1,17 @@
+"""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