robocasa.environments.kitchen.multi_stage.clearing_table package

Submodules

robocasa.environments.kitchen.multi_stage.clearing_table.bowl_and_cup module

class robocasa.environments.kitchen.multi_stage.clearing_table.bowl_and_cup.BowlAndCup(*args, **kwargs)

Bases: Kitchen

Bowl And Cup: composite task for Clearing Table activity.

Simulates the process of efficiently clearing the table.

Steps:

Place the cup inside the bowl on the island and move it to any counter.

Restricted to layouts with an island.

EXCLUDE_LAYOUTS = [0, 2, 4, 5, 7, 8, 9]

get_ep_meta()

Returns a dictionary containing episode meta data

robocasa.environments.kitchen.multi_stage.clearing_table.candle_cleanup module

class robocasa.environments.kitchen.multi_stage.clearing_table.candle_cleanup.CandleCleanup(cab_id=FixtureType.CABINET_TOP, *args, **kwargs)

Bases: Kitchen

Candle Cleanup: composite task for Clearing Table activity.

Simulates the process of efficiently clearing the dining table decorations.

Steps:

Pick the decorations from the dining table and place it in the open cabinet.

Parameters:

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

EXCLUDE_LAYOUTS = [0, 2, 4, 5]

get_ep_meta()

Returns a dictionary containing episode meta data

robocasa.environments.kitchen.multi_stage.clearing_table.clearing_cleaning_receptacles module

class robocasa.environments.kitchen.multi_stage.clearing_table.clearing_cleaning_receptacles.ClearingCleaningReceptacles(*args, **kwargs)

Bases: Kitchen

Clearing Cleaning Receptacles: composite task for Clearing Table activity.

Simulates the process of clearing receptacles from the dining table and cleaning them in the sink.

Steps:

Pick the receptacles from the dining table and place them in the sink. Then, turn on the water.

EXCLUDE_LAYOUTS = [0, 2, 4, 5]

get_ep_meta()

Returns a dictionary containing episode meta data

robocasa.environments.kitchen.multi_stage.clearing_table.condiment_collection module

class robocasa.environments.kitchen.multi_stage.clearing_table.condiment_collection.CondimentCollection(cab_id=FixtureType.CABINET_TOP, *args, **kwargs)

Bases: Kitchen

Condiment Collection: composite task for Clearing Table activity.

Simulates the task of collecting condiments from the counter and placing them in the cabinet.

Steps:

Pick the condiments from the counter and place them in the cabinet.

Parameters:

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

get_ep_meta()

Returns a dictionary containing episode meta data

robocasa.environments.kitchen.multi_stage.clearing_table.dessert_assembly module

class robocasa.environments.kitchen.multi_stage.clearing_table.dessert_assembly.DessertAssembly(*args, **kwargs)

Bases: Kitchen

Dessert Assembly: composite task for Clearing Table activity.

Simulates the task of assembling desserts.

Steps:

Pick the container with the dessert on it and place in on the tray. Pick the cupcake and place it on the tray.

get_ep_meta()

Returns a dictionary containing episode meta data

robocasa.environments.kitchen.multi_stage.clearing_table.drinkware_consolidation module

class robocasa.environments.kitchen.multi_stage.clearing_table.drinkware_consolidation.DrinkwareConsolidation(*args, **kwargs)

Bases: Kitchen

Drinkware Consolidation: composite task for Clearing Table activity.

Simulates the task of clearing the island drinkware and placing them back in a cabinet.

Steps:

Pick the drinkware from the island and place them in the open cabinet.

EXCLUDE_LAYOUTS = [0, 2, 4, 5, 7, 8, 9]

get_ep_meta()

Returns a dictionary containing episode meta data

reset()

Resets simulation. :returns: Environment observation space after reset occurs :rtype: OrderedDict

robocasa.environments.kitchen.multi_stage.clearing_table.food_cleanup module

class robocasa.environments.kitchen.multi_stage.clearing_table.food_cleanup.FoodCleanup(cab_id=FixtureType.CABINET_TOP, *args, **kwargs)

Bases: Kitchen

Food Cleanup: composite task for Clearing Table activity.

Simulates the task of cleaning up various food items left on the counter.

Steps:

Pick the food items from the counter and place them in the cabinet. Then close the cabinet.

Parameters:

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

get_ep_meta()

Returns a dictionary containing episode meta data

Module contents