Skip to content

Commit

Permalink
more update on task notes
Browse files Browse the repository at this point in the history
  • Loading branch information
lf-zhao committed Nov 5, 2024
1 parent bea44c7 commit c88f5f9
Showing 1 changed file with 29 additions and 15 deletions.
44 changes: 29 additions & 15 deletions predicators/envs/spot_env.py
Original file line number Diff line number Diff line change
Expand Up @@ -3307,30 +3307,43 @@ def _get_dry_task(self, train_or_test: str,
# LIS Envs and Tasks #
###############################################################################

# Tasks
"""
# LISSpotBlockFloorEnv - pick only, FO
Tasks:
### 1. Fully Observable (FO) Tasks
# Pick Only
LISSpotBlockFloorEnv:
- "pick up the red block"
# LISSpotBlockBowlEnv - pick place, FO
# Pick and Place
LISSpotBlockBowlEnv:
- "pick the red block into the green bowl"
# LISSpotBlockInBoxEnv - pick place, FO
LISSpotBlockInBoxEnv:
- "put the red block into the cardboard box on floor"
- "view the object from top"
- "view the object from top" # Verify info gathering operator
### 2. Partially Observable (PO) Tasks with Information Gathering
NOTE: PO tasks require specialized operators (e.g., MoveToHandObserveObjectFromTop) to handle information gathering
about object states before manipulation.
# Pick Only with Info Gathering
LISSpotEmptyCupPickEnv:
- "know whether the cup is empty" # Info gathering goal
- "pick up the empty cup" # Main goal
# Pick and Place with Info Gathering
LISSpotEmptyCupBoxEnv:
- "place empty cup into the box" # Requires verifying cup is empty first
NOTE: Partially Observable (PO) tasks need specialized operators to handle
information gathering.
# - pick only, PO with info gathering
- "know container not as empty" / "know whether the cup is empty"
- "pick up the empty cup"
LISSpotTableCupInBoxEnv:
- "put the cup into the cardboard box on floor" # Requires checking cup state
# LISSpotTableCupInBoxEnv - pick place, PO with info gathering
- "put the cup into the cardboard box on floor"
- "place empty cup into the box"
LISSpotBlockTableInBowlEnv:
- "put the red block on table into the green bowl on floor" # Requires checking table state
# LISSpotBlockTableInBowlEnv - pick place, PO with info gathering
- "put the red block on table into the green bowl on floor"
# Pick Place in more open space
- TODO
"""


Expand All @@ -3345,6 +3358,7 @@ def _get_dry_task(self, train_or_test: str,

# Movable objects: cups, potentially with lids and non-empty
"cup": "orange cup/orange cylinder/orange-ish mug",
"orange_cup": "orange cup/orange cylinder/orange-ish mug",
"blue_cup": "blue cup/blue mug/uncovered blue cup",
"green_cup": "green cup/greenish cup/green cylinder",

Expand Down

0 comments on commit c88f5f9

Please sign in to comment.