TASK_DEFINITION = { 1: { "name": "Single Item Retrieval", "level": 1, "targets": [[5, 5]], "max_steps": 50, "description": "Navigate to [5,5], pick the item, and return to the Dock at [0,0]." }, 2: { "name": "Multi-Order Sequential", "level": 2, "targets": [[8, 2], [2, 8]], "max_steps": 150, "description": "Collect two items in order and return each safely to the Dock." }, 3: { "name": "Dynamic Warehouse Master", "level": 3, "targets": [[9, 9], [5, 1], [1, 5]], "max_steps": 300, "description": "Avoid moving forklifts while fulfilling a triple-item order." } } def get_task(task_id): return TASK_DEFINITION.get(task_id, TASK_DEFINITION[1])