robocasa.environments.kitchen.single_stage package
Contents
robocasa.environments.kitchen.single_stage package¶
Submodules¶
robocasa.environments.kitchen.single_stage.kitchen_coffee module¶
- class robocasa.environments.kitchen.single_stage.kitchen_coffee.CoffeePressButton(*args, **kwargs)¶
Bases:
Kitchen
Class encapsulating the coffee press button task. Press the button on the coffee machine to serve coffee.
- get_ep_meta()¶
Get the episode metadata for the coffee press button task. This includes the language description of the task.
- Returns:
Episode metadata.
- Return type:
dict
- class robocasa.environments.kitchen.single_stage.kitchen_coffee.CoffeeServeMug(*args, **kwargs)¶
Bases:
PnPCoffee
Class encapsulating the coffee serve task. Pick the mug from under the coffee machine dispenser and place it on the counter.
- class robocasa.environments.kitchen.single_stage.kitchen_coffee.CoffeeSetupMug(*args, **kwargs)¶
Bases:
PnPCoffee
Class encapsulating the coffee setup task. Pick the mug from the counter and place it under the coffee machine dispenser.
- class robocasa.environments.kitchen.single_stage.kitchen_coffee.PnPCoffee(behavior='machine_to_counter', *args, **kwargs)¶
Bases:
Kitchen
Class encapsulating the atomic pick and place coffee tasks.
- Parameters:
behavior (str) – “counter_to_machine” or “machine_to_counter”. Used to define the desired pick and place behavior for the task.
- get_ep_meta()¶
Get the episode metadata for the coffee tasks. This includes the language description of the task.
- Returns:
Episode metadata.
- Return type:
dict
robocasa.environments.kitchen.single_stage.kitchen_doors module¶
- class robocasa.environments.kitchen.single_stage.kitchen_doors.CloseDoor(behavior=None, *args, **kwargs)¶
Bases:
ManipulateDoor
- class robocasa.environments.kitchen.single_stage.kitchen_doors.CloseDoubleDoor(door_id=FixtureType.DOOR_TOP_HINGE_DOUBLE, *args, **kwargs)¶
Bases:
CloseDoor
- class robocasa.environments.kitchen.single_stage.kitchen_doors.CloseSingleDoor(door_id=FixtureType.DOOR_TOP_HINGE_SINGLE, *args, **kwargs)¶
Bases:
CloseDoor
- class robocasa.environments.kitchen.single_stage.kitchen_doors.ManipulateDoor(behavior='open', door_id=FixtureType.DOOR_TOP_HINGE, *args, **kwargs)¶
Bases:
Kitchen
Class encapsulating the atomic manipulate door tasks.
- Parameters:
behavior (str) – “open” or “close”. Used to define the desired door manipulation behavior for the task.
door_id (str) – The door fixture id to manipulate.
- get_ep_meta()¶
Get the episode metadata for the door tasks. This includes the language description of the task.
- Returns:
Episode metadata.
- Return type:
dict
- class robocasa.environments.kitchen.single_stage.kitchen_doors.OpenDoor(*args, **kwargs)¶
Bases:
ManipulateDoor
robocasa.environments.kitchen.single_stage.kitchen_drawer module¶
- class robocasa.environments.kitchen.single_stage.kitchen_drawer.CloseDrawer(*args, **kwargs)¶
Bases:
ManipulateDrawer
Class encapsulating the atomic close drawer task.
- class robocasa.environments.kitchen.single_stage.kitchen_drawer.ManipulateDrawer(behavior='open', drawer_id=FixtureType.TOP_DRAWER, *args, **kwargs)¶
Bases:
Kitchen
Class encapsulating the atomic manipulate drawer tasks.
- Parameters:
behavior (str) – “open” or “close”. Used to define the desired drawer manipulation behavior for the task.
drawer_id (str) – The drawer fixture id to manipulate
- check_fxtr_contact(pos)¶
Check if the point is in contact with any fixture
- Parameters:
pos (tuple) – The position of the point to check
- Returns:
True if the point is in contact with any fixture, False otherwise
- Return type:
bool
- check_sidewall_contact(pos)¶
Check if the point is in contact with any wall. This is separate from check_fxtr_contact because the point could be outside of the wall’s bounds but still in contact with the wall since the wall is thin.
- Parameters:
pos (tuple) – The position of the point to check
- Returns:
True if the point is in contact with any wall, False otherwise
- Return type:
bool
- get_ep_meta()¶
Get the episode metadata for the drawer tasks. This includes the language description of the task.
- class robocasa.environments.kitchen.single_stage.kitchen_drawer.OpenDrawer(*args, **kwargs)¶
Bases:
ManipulateDrawer
Class encapsulating the atomic open drawer task.
robocasa.environments.kitchen.single_stage.kitchen_microwave module¶
- class robocasa.environments.kitchen.single_stage.kitchen_microwave.MicrowavePressButton(behavior='turn_on', *args, **kwargs)¶
Bases:
Kitchen
Class encapsulating the atomic microwave press button tasks.
- Parameters:
behavior (str) – “turn_on” or “turn_off”. Used to define the desired microwave manipulation behavior for the task
- get_ep_meta()¶
Get the episode metadata for the microwave tasks. This includes the language description of the task.
- class robocasa.environments.kitchen.single_stage.kitchen_microwave.TurnOffMicrowave(behavior=None, *args, **kwargs)¶
Bases:
MicrowavePressButton
- class robocasa.environments.kitchen.single_stage.kitchen_microwave.TurnOnMicrowave(behavior=None, *args, **kwargs)¶
Bases:
MicrowavePressButton
robocasa.environments.kitchen.single_stage.kitchen_pnp module¶
- class robocasa.environments.kitchen.single_stage.kitchen_pnp.PnP(obj_groups='all', exclude_obj_groups=None, *args, **kwargs)¶
Bases:
Kitchen
Class encapsulating the atomic pick and place tasks.
- Parameters:
obj_groups (str) – Object groups to sample the target object from.
exclude_obj_groups (str) – Object groups to exclude from sampling the target object.
- class robocasa.environments.kitchen.single_stage.kitchen_pnp.PnPCabToCounter(cab_id=FixtureType.CABINET_TOP, obj_groups='all', *args, **kwargs)¶
Bases:
PnP
Class encapsulating the atomic cabinet to counter pick and place task
- Parameters:
cab_id (str) – The cabinet fixture id to pick the object from.
obj_groups (str) – Object groups to sample the target object from.
- get_ep_meta()¶
Get the episode metadata for the cabinet to counter pick and place task. This includes the language description of the task.
- class robocasa.environments.kitchen.single_stage.kitchen_pnp.PnPCounterToCab(cab_id=FixtureType.CABINET_TOP, obj_groups='all', *args, **kwargs)¶
Bases:
PnP
Class encapsulating the atomic counter to cabinet pick and place task
- Parameters:
cab_id (str) – The cabinet fixture id to place the object.
obj_groups (str) – Object groups to sample the target object from.
- get_ep_meta()¶
Get the episode metadata for the counter to cabinet pick and place task. This includes the language description of the task.
- class robocasa.environments.kitchen.single_stage.kitchen_pnp.PnPCounterToMicrowave(obj_groups='food', *args, **kwargs)¶
Bases:
PnP
- EXCLUDE_LAYOUTS = [8]¶
Class encapsulating the atomic counter to microwave pick and place task
- Parameters:
obj_groups (str) – Object groups to sample the target object from.
- get_ep_meta()¶
Get the episode metadata for the counter to microwave pick and place task. This includes the language description of the task.
- class robocasa.environments.kitchen.single_stage.kitchen_pnp.PnPCounterToSink(obj_groups='all', *args, **kwargs)¶
Bases:
PnP
Class encapsulating the atomic counter to sink pick and place task
- Parameters:
obj_groups (str) – Object groups to sample the target object from.
- get_ep_meta()¶
Get the episode metadata for the counter to sink pick and place task. This includes the language description of the task.
- class robocasa.environments.kitchen.single_stage.kitchen_pnp.PnPCounterToStove(obj_groups='food', *args, **kwargs)¶
Bases:
PnP
Class encapsulating the atomic counter to stove pick and place task
- Parameters:
obj_groups (str) – Object groups to sample the target object from.
- get_ep_meta()¶
Get the episode metadata for the counter to stove pick and place task. This includes the language description of the task.
- class robocasa.environments.kitchen.single_stage.kitchen_pnp.PnPMicrowaveToCounter(obj_groups='food', *args, **kwargs)¶
Bases:
PnP
- EXCLUDE_LAYOUTS = [8]¶
Class encapsulating the atomic microwave to counter pick and place task
- Parameters:
obj_groups (str) – Object groups to sample the target object from.
- get_ep_meta()¶
Get the episode metadata for the microwave to counter pick and place task. This includes the language description of the task.
- class robocasa.environments.kitchen.single_stage.kitchen_pnp.PnPSinkToCounter(obj_groups='food', *args, **kwargs)¶
Bases:
PnP
Class encapsulating the atomic sink to counter pick and place task
- Parameters:
obj_groups (str) – Object groups to sample the target object from.
- get_ep_meta()¶
Get the episode metadata for the sink to counter pick and place task. This includes the language description of the task.
- class robocasa.environments.kitchen.single_stage.kitchen_pnp.PnPStoveToCounter(obj_groups='food', *args, **kwargs)¶
Bases:
PnP
Class encapsulating the atomic stove to counter pick and place task
- get_ep_meta()¶
Get the episode metadata for the stove to counter pick and place task. This includes the language description of the task.
robocasa.environments.kitchen.single_stage.kitchen_sink module¶
- class robocasa.environments.kitchen.single_stage.kitchen_sink.ManipulateSinkFaucet(behavior='turn_on', *args, **kwargs)¶
Bases:
Kitchen
Class encapsulating the atomic manipulate sink faucet tasks.
- Parameters:
behavior (str) – “turn_on” or “turn_off”. Used to define the desired sink faucet manipulation behavior for the task.
- get_ep_meta()¶
Get the episode metadata for the sink faucet tasks. This includes the language description of the task.
- class robocasa.environments.kitchen.single_stage.kitchen_sink.TurnOffSinkFaucet(*args, **kwargs)¶
Bases:
ManipulateSinkFaucet
- class robocasa.environments.kitchen.single_stage.kitchen_sink.TurnOnSinkFaucet(*args, **kwargs)¶
Bases:
ManipulateSinkFaucet
- class robocasa.environments.kitchen.single_stage.kitchen_sink.TurnSinkSpout(*args, **kwargs)¶
Bases:
Kitchen
Class encapsulating the atomic turn sink spout tasks.
- Parameters:
behavior (str) – “left” or “right”. Used to define the desired sink spout
task. (manipulation behavior for the)
- get_ep_meta()¶
Get the episode metadata for the sink spout tasks. This includes the language description of the task.
robocasa.environments.kitchen.single_stage.kitchen_stove module¶
- class robocasa.environments.kitchen.single_stage.kitchen_stove.ManipulateStoveKnob(knob_id='random', behavior='turn_on', *args, **kwargs)¶
Bases:
Kitchen
Class encapsulating the atomic manipulate stove knob tasks.
- Parameters:
knob_id (str) – The stove knob id to manipulate. If set to “random”, a random knob will be selected.
behavior (str) – “turn_on” or “turn_off”. Used to define the desired stove knob manipulation behavior for the task.
- get_ep_meta()¶
Get the episode metadata for the stove knob tasks. This includes the language description of the task and the task references.
- class robocasa.environments.kitchen.single_stage.kitchen_stove.TurnOffStove(*args, **kwargs)¶
Bases:
ManipulateStoveKnob
- class robocasa.environments.kitchen.single_stage.kitchen_stove.TurnOnStove(*args, **kwargs)¶
Bases:
ManipulateStoveKnob