making more test tasks for cooking

This commit is contained in:
Isadora White 2025-05-10 18:17:06 -07:00
parent 4ae95cba38
commit e049abb708
3 changed files with 2134 additions and 16 deletions

View file

@ -257,7 +257,7 @@ def count_items_in_inventory(inventory):
def reconfigure_tasks(task_path, new_task_path, num_agents=None): def reconfigure_tasks(task_path, new_task_path, num_agents=None, hells_kitchen=False):
with open(task_path, 'r') as f: with open(task_path, 'r') as f:
tasks = json.load(f) tasks = json.load(f)
task_ids = tasks.keys() task_ids = tasks.keys()
@ -313,17 +313,18 @@ def reconfigure_tasks(task_path, new_task_path, num_agents=None):
task["initial_inventory"] = initial_inventory task["initial_inventory"] = initial_inventory
print(inventory) print(inventory)
print(initial_inventory) print(initial_inventory)
goals = task.get("goal", {}) if not hells_kitchen:
new_goals = {} goals = task.get("goal", {})
blocked_access = task.get("blocked_access_to_recipe", []) new_goals = {}
print(blocked_access) blocked_access = task.get("blocked_access_to_recipe", [])
for key, goal in goals.items(): print(blocked_access)
initial_goal = goal.split("\n")[0] for key, goal in goals.items():
if str(key) not in blocked_access: initial_goal = goal.split("\n")[0]
for item, recipe in new_recipes.items(): if str(key) not in blocked_access:
initial_goal += f"Recipe for {item}:\n{recipe}" for item, recipe in new_recipes.items():
new_goals[key] = initial_goal initial_goal += f"Recipe for {item}:\n{recipe}"
task["goal"] = new_goals new_goals[key] = initial_goal
task["goal"] = new_goals
new_tasks[task_id] = task new_tasks[task_id] = task
# check each of the recipes and replace with the new recipe # check each of the recipes and replace with the new recipe
@ -342,10 +343,11 @@ def reconfigure_tasks(task_path, new_task_path, num_agents=None):
# reconfigure_tasks("mindcraft/tasks/cooking_tasks/test_tasks/2_agent_cooking_test_tasks.json", "mindcraft/tasks/cooking_tasks/require_collab_test_2_items/3_agent.json", 3) # reconfigure_tasks("mindcraft/tasks/cooking_tasks/test_tasks/2_agent_cooking_test_tasks.json", "mindcraft/tasks/cooking_tasks/require_collab_test_2_items/3_agent.json", 3)
# reconfigure_tasks("mindcraft/tasks/cooking_tasks/test_tasks/2_agent_cooking_test_tasks.json", "mindcraft/tasks/cooking_tasks/require_collab_test_2_items/4_agent.json", 4) # reconfigure_tasks("mindcraft/tasks/cooking_tasks/test_tasks/2_agent_cooking_test_tasks.json", "mindcraft/tasks/cooking_tasks/require_collab_test_2_items/4_agent.json", 4)
# reconfigure_tasks("mindcraft/tasks/cooking_tasks/test_tasks/2_agent_cooking_test_tasks.json", "mindcraft/tasks/cooking_tasks/require_collab_test_2_items/5_agent.json", 5) # reconfigure_tasks("mindcraft/tasks/cooking_tasks/test_tasks/2_agent_cooking_test_tasks.json", "mindcraft/tasks/cooking_tasks/require_collab_test_2_items/5_agent.json", 5)
reconfigure_tasks("mindcraft/tasks/cooking_tasks/train_tasks/2_agent_cooking_train_tasks.json", "mindcraft/tasks/cooking_tasks/require_collab_train_2_items/2_agent.json", 2) reconfigure_tasks("mindcraft/tasks/cooking_tasks/test_tasks/test_tasks.json", "mindcraft/tasks/cooking_tasks/require_collab_train_2_items/2_agent_block_recipe.json", 2)
reconfigure_tasks("mindcraft/tasks/cooking_tasks/train_tasks/2_agent_cooking_train_tasks.json", "mindcraft/tasks/cooking_tasks/require_collab_train_2_items/3_agent.json", 3) reconfigure_tasks("mindcraft/tasks/cooking_tasks/test_tasks/hells_kitchen_test_tasks.json", "mindcraft/tasks/cooking_tasks/require_collab_train_2_items/2_agent_hells_kitchen.json", 2, True)
reconfigure_tasks("mindcraft/tasks/cooking_tasks/train_tasks/2_agent_cooking_train_tasks.json", "mindcraft/tasks/cooking_tasks/require_collab_train_2_items/4_agent.json", 4) # reconfigure_tasks("mindcraft/tasks/cooking_tasks/train_tasks/2_agent_cooking_train_tasks.json", "mindcraft/tasks/cooking_tasks/require_collab_train_2_items/3_agent.json", 3)
reconfigure_tasks("mindcraft/tasks/cooking_tasks/train_tasks/2_agent_cooking_train_tasks.json", "mindcraft/tasks/cooking_tasks/require_collab_train_2_items/5_agent.json", 5) # reconfigure_tasks("mindcraft/tasks/cooking_tasks/train_tasks/2_agent_cooking_train_tasks.json", "mindcraft/tasks/cooking_tasks/require_collab_train_2_items/4_agent.json", 4)
# reconfigure_tasks("mindcraft/tasks/cooking_tasks/train_tasks/2_agent_cooking_train_tasks.json", "mindcraft/tasks/cooking_tasks/require_collab_train_2_items/5_agent.json", 5)
# reconfigure_tasks("mindcraft/tasks/cooking_tasks/equal_load_test_tasks/3_agent.json", "mindcraft/tasks/cooking_tasks/require_collab_test/3_agent.json") # reconfigure_tasks("mindcraft/tasks/cooking_tasks/equal_load_test_tasks/3_agent.json", "mindcraft/tasks/cooking_tasks/require_collab_test/3_agent.json")
# reconfigure_tasks("mindcraft/tasks/cooking_tasks/equal_load_test_tasks/4_agent.json", "mindcraft/tasks/cooking_tasks/require_collab_test/4_agent.json") # reconfigure_tasks("mindcraft/tasks/cooking_tasks/equal_load_test_tasks/4_agent.json", "mindcraft/tasks/cooking_tasks/require_collab_test/4_agent.json")
# reconfigure_tasks("mindcraft/tasks/cooking_tasks/equal_load_test_tasks/5_agent.json", "mindcraft/tasks/cooking_tasks/require_collab_test/5_agent.json") # reconfigure_tasks("mindcraft/tasks/cooking_tasks/equal_load_test_tasks/5_agent.json", "mindcraft/tasks/cooking_tasks/require_collab_test/5_agent.json")

File diff suppressed because it is too large Load diff

File diff suppressed because it is too large Load diff