robocasa.environments.kitchen.multi_stage.setting_the_table package

Submodules

robocasa.environments.kitchen.multi_stage.setting_the_table.arrange_bread_basket module

class robocasa.environments.kitchen.multi_stage.setting_the_table.arrange_bread_basket.ArrangeBreadBasket(cab_id=FixtureType.CABINET_TOP, *args, **kwargs)

Bases: Kitchen

Arrange Bread Basket: composite task for Setting The Table activity.

Simulates the task of arranging the bread basket.

Steps:

Pick the bread from the cabinet and place it in the bowl. Then move the bowl to the dining counter.

Restricted to layouts which have a dining table (long counter area with stools).

Parameters:

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

EXCLUDE_LAYOUTS = [0, 2, 4, 5]
get_ep_meta()

Returns a dictionary containing episode meta data

robocasa.environments.kitchen.multi_stage.setting_the_table.beverage_organization module

class robocasa.environments.kitchen.multi_stage.setting_the_table.beverage_organization.BeverageOrganization(*args, **kwargs)

Bases: Kitchen

Beverage Organization: composite task for Setting The Table activity.

Simulates the task of organizing beverages.

Steps:

Move the drinks to the dining counter.

Restricted to layouts which have a dining table (long counter area with stools).

EXCLUDE_LAYOUTS = [0, 2, 4, 5]
get_ep_meta()

Returns a dictionary containing episode meta data

robocasa.environments.kitchen.multi_stage.setting_the_table.date_night module

class robocasa.environments.kitchen.multi_stage.setting_the_table.date_night.DateNight(cab_id=FixtureType.CABINET_TOP, *args, **kwargs)

Bases: Kitchen

Date Night: composite task for Setting The Table activity.

Simulates the task of setting the table for a date night.

Steps:

Pick up the decoration and the alcohol from the cabinet and move them to the dining counter.

Restricted to layouts which have a dining table (long counter area with stools).

Parameters:

cab_id (int) – Enum which serves as a unique identifier for different cabinet types. Used to choose the cabinet from which the decoration and alcohol are picked.

EXCLUDE_LAYOUTS = [0, 2, 4, 5]
get_ep_meta()

Returns a dictionary containing episode meta data

robocasa.environments.kitchen.multi_stage.setting_the_table.seasoning_spice_setup module

class robocasa.environments.kitchen.multi_stage.setting_the_table.seasoning_spice_setup.SeasoningSpiceSetup(cab_id=FixtureType.CABINET_TOP, *args, **kwargs)

Bases: Kitchen

Seasoning Spice Setup: composite task for Setting The Table activity.

Simulates the task of setting the table with seasoning and spices.

Steps:

Move the seasoning and spices from the cabinet directly in front to the dining counter.

Restricted to layouts which have a dining table (long counter area with stools).

Parameters:

cab_id (int) – Enum which serves as a unique identifier for different cabinet types. Used to choose the cabinet from which the seasoning and spices are picked.

EXCLUDE_LAYOUTS = [0, 2, 4, 5]
get_ep_meta()

Returns a dictionary containing episode meta data

robocasa.environments.kitchen.multi_stage.setting_the_table.set_bowls_for_soup module

class robocasa.environments.kitchen.multi_stage.setting_the_table.set_bowls_for_soup.SetBowlsForSoup(cab_id=FixtureType.DOOR_TOP_HINGE_DOUBLE, *args, **kwargs)

Bases: Kitchen

Set Bowls For Soup: composite task for Setting The Table activity.

Simulates the task of setting the table with bowls for soup.

Steps:

Move the bowls from the cabinet to the plates on the dining table.

Restricted to layouts which have a dining table (long counter area with stools).

Parameters:

cab_id (int) – Enum which serves as a unique identifier for different cabinet types. Used to choose the cabinet from which the bowls are picked.

EXCLUDE_LAYOUTS = [0, 2, 4, 5]
get_ep_meta()

Returns a dictionary containing episode meta data

robocasa.environments.kitchen.multi_stage.setting_the_table.size_sorting module

class robocasa.environments.kitchen.multi_stage.setting_the_table.size_sorting.SizeSorting(*args, **kwargs)

Bases: Kitchen

Size Sorting: composite task for Setting The Table activity.

Simulates the task of stacking objects by size.

Steps:

Stack the objects from largest to smallest.

get_ep_meta()

Returns a dictionary containing episode meta data

Module contents