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: robocasa.environments.kitchen.kitchen.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: robocasa.environments.kitchen.kitchen.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: robocasa.environments.kitchen.kitchen.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: robocasa.environments.kitchen.kitchen.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#