robocasa.environments.kitchen.multi_stage.restocking_supplies package

Submodules

robocasa.environments.kitchen.multi_stage.restocking_supplies.beverage_sorting module

class robocasa.environments.kitchen.multi_stage.restocking_supplies.beverage_sorting.BeverageSorting(*args, **kwargs)

Bases: Kitchen

Beverage Sorting: composite task for Restocking Supplies activity.

Simulates the task of sorting beverages.

Steps:

Sort all alcoholic drinks to one cabinet, and non-alcoholic drinks to the other.

get_ep_meta()

Returns a dictionary containing episode meta data

robocasa.environments.kitchen.multi_stage.restocking_supplies.restock_bowls module

class robocasa.environments.kitchen.multi_stage.restocking_supplies.restock_bowls.RestockBowls(cab_id=FixtureType.DOOR_TOP_HINGE_DOUBLE, *args, **kwargs)

Bases: Kitchen

Restock Bowls: composite task for Restocking Supplies activity.

Simulates the task of restocking bowls.

Steps:

Restock two bowls from the counter to the cabinet.

Parameters:

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

get_ep_meta()

Returns a dictionary containing episode meta data

robocasa.environments.kitchen.multi_stage.restocking_supplies.restock_pantry module

class robocasa.environments.kitchen.multi_stage.restocking_supplies.restock_pantry.RestockPantry(*args, **kwargs)

Bases: Kitchen

Restock Pantry: composite task for Restocking Supplies activity.

Simulates the task of organizing cans when restocking them.

Steps:

Pick the cans from the counter and place them on the side of the cabinet that already has a can.

get_ep_meta()

Returns a dictionary containing episode meta data

robocasa.environments.kitchen.multi_stage.restocking_supplies.stocking_breakfast_foods module

class robocasa.environments.kitchen.multi_stage.restocking_supplies.stocking_breakfast_foods.StockingBreakfastFoods(cab_id1=FixtureType.CABINET_TOP, cab_id2=FixtureType.CABINET_TOP, *args, **kwargs)

Bases: Kitchen

Stocking Breakfast Foods: composite task for Restocking Supplies activity.

Simulates the task of restocking breakfast foods.

Steps:

Restock two packaged foods from the counter to the cabinet.

Parameters:

  • cab_id1 (int) – Enum which serves as a unique identifier for different cabinet types. Used to choose the first cabinet to which the foods are restocked.

  • cab_id2 (int) – Enum which serves as a unique identifier for different cabinet types. Used to choose the second cabinet to which the foods are restocked.

get_ep_meta()

Returns a dictionary containing episode meta data

Module contents