File size: 648 Bytes
987ed1b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 | from diffusion_policy.env.kitchen.base import KitchenBase
class KitchenMicrowaveKettleBottomBurnerLightV0(KitchenBase):
TASK_ELEMENTS = ["microwave", "kettle", "bottom burner", "light switch"]
COMPLETE_IN_ANY_ORDER = False
class KitchenMicrowaveKettleLightSliderV0(KitchenBase):
TASK_ELEMENTS = ["microwave", "kettle", "light switch", "slide cabinet"]
COMPLETE_IN_ANY_ORDER = False
class KitchenKettleMicrowaveLightSliderV0(KitchenBase):
TASK_ELEMENTS = ["kettle", "microwave", "light switch", "slide cabinet"]
COMPLETE_IN_ANY_ORDER = False
class KitchenAllV0(KitchenBase):
TASK_ELEMENTS = KitchenBase.ALL_TASKS
|