robocasa.environments.kitchen.multi_stage.baking package

Submodules

robocasa.environments.kitchen.multi_stage.baking.cupcake_cleanup module

class robocasa.environments.kitchen.multi_stage.baking.cupcake_cleanup.CupcakeCleanup(*args, **kwargs)

Bases: robocasa.environments.kitchen.kitchen.Kitchen

Cupcake Cleanup: composite task for Baking activity.

Simulates the task of cleaning up after baking cupcakes.

Steps:

Move the cupcake off the tray onto the counter, and place the bowl used for mixing into the sink.

get_ep_meta()

Returns a dictionary containing episode meta data

robocasa.environments.kitchen.multi_stage.baking.organize_baking_ingredients module

class robocasa.environments.kitchen.multi_stage.baking.organize_baking_ingredients.OrganizeBakingIngredients(*args, **kwargs)

Bases: robocasa.environments.kitchen.kitchen.Kitchen

Organize Baking Ingredients: composite task for Baking activity.

Simulates the task of organizing baking ingredients.

Steps:

Place the eggs and milk near the bowl on the counter.

get_ep_meta()

Returns a dictionary containing episode meta data

robocasa.environments.kitchen.multi_stage.baking.pastry_display module

class robocasa.environments.kitchen.multi_stage.baking.pastry_display.PastryDisplay(*args, **kwargs)

Bases: robocasa.environments.kitchen.kitchen.Kitchen

Pastry Display: composite task for Baking activity.

Simulates the task of displaying pastries.

Steps:

Place the pastries on the plates.

get_ep_meta()

Returns a dictionary containing episode meta data

Module contents