robocasa.environments.kitchen.multi_stage.tidying_cabinets_and_drawers package

Submodules

robocasa.environments.kitchen.multi_stage.tidying_cabinets_and_drawers.drawer_utensil_sort module

class robocasa.environments.kitchen.multi_stage.tidying_cabinets_and_drawers.drawer_utensil_sort.DrawerUtensilSort(drawer_id=FixtureType.TOP_DRAWER, *args, **kwargs)

Bases: ManipulateDrawer

Drawer Utensil Sort: composite task for Tidying Cabinets And Drawers activity.

Simulates the task of placing utensils in the drawer.

Steps:

Open the drawer and push the utensils inside it.

Parameters:

drawer_id (int) – Enum which serves as a unique identifier for different drawer types. Used to choose the drawer in which the utensils are placed. TOP_DRAWER specifies the uppermost drawer.

get_ep_meta()

Get the episode metadata for the drawer tasks. This includes the language description of the task.

robocasa.environments.kitchen.multi_stage.tidying_cabinets_and_drawers.organize_cleaning_supplies module

class robocasa.environments.kitchen.multi_stage.tidying_cabinets_and_drawers.organize_cleaning_supplies.OrganizeCleaningSupplies(cab_id=FixtureType.CABINET_TOP, *args, **kwargs)

Bases: Kitchen

Organize Cleaning Supplies: composite task for Tidying Cabinets And Drawers activity.

Simulates the task of preparing to clean the sink.

Steps:

Open the cabinet. Pick the cleaner and place it next to the sink. Then close the cabinet.

Parameters:

cab_id (str) – Enum which serves as a unique identifier for different cabinet types. Used to choose the cabinet from which the cleaner is picked.

get_ep_meta()

Returns a dictionary containing episode meta data

robocasa.environments.kitchen.multi_stage.tidying_cabinets_and_drawers.pantry_mishap module

class robocasa.environments.kitchen.multi_stage.tidying_cabinets_and_drawers.pantry_mishap.PantryMishap(*args, **kwargs)

Bases: ManipulateDrawer

Pantry Mishap: composite task for Tidying Cabinets And Drawers activity.

Simulates the task of organizing the pantry after a mishap from the incorrect placement of items in the cabinet.

Steps:

Open the cabinet. Pick the vegetable and place it on the counter. Pick the canned food and place it in the drawer. Close the cabinet.

get_ep_meta()

Get the episode metadata for the drawer tasks. This includes the language description of the task.

robocasa.environments.kitchen.multi_stage.tidying_cabinets_and_drawers.shaker_shuffle module

class robocasa.environments.kitchen.multi_stage.tidying_cabinets_and_drawers.shaker_shuffle.ShakerShuffle(*args, **kwargs)

Bases: ManipulateDrawer

Shaker Shuffle: composite task for Tidying Cabinets And Drawers activity.

Simulates the task of reorganizing the pantry by only placing the shakers in the drawer.

Steps:

Open the cabinet. Place the shakers in the drawer. Close the cabinet.

get_ep_meta()

Get the episode metadata for the drawer tasks. This includes the language description of the task.

robocasa.environments.kitchen.multi_stage.tidying_cabinets_and_drawers.snack_sorting module

class robocasa.environments.kitchen.multi_stage.tidying_cabinets_and_drawers.snack_sorting.SnackSorting(*args, **kwargs)

Bases: ManipulateDrawer

Snack Sorting: composite task for Tidying Cabinets And Drawers activity.

Simulates the task of placing snacks in the bowl.

Steps:

Place the bar in the bowl and close the drawer.

get_ep_meta()

Get the episode metadata for the drawer tasks. This includes the language description of the task.

Module contents