plancraft 0.4.5__py3-none-any.whl → 0.4.7__py3-none-any.whl

plancraft/environment/planner.py

@@ -392,7 +392,8 @@ def decompose_subgoal(
 
                     # if item is already in the correct position, skip
                     if (
-                        inventory_position in current_inventory
+                        items_to_use_counter[item] > 0
+                        and inventory_position in current_inventory
                         and current_inventory[inventory_position]["type"] == item
                     ) and current_inventory[inventory_position]["quantity"] > 0:
                         min_quantity_needed.add(inventory_position)
@@ -441,6 +442,8 @@ def decompose_subgoal(
                         )
                         action_items_used.append(current_inventory[from_slot]["type"])
                         items_to_use_counter[item] -= 1
+                        if items_to_use_counter[item] == 0:
+                            del items_to_use_counter[item]
                         added_item = True
                         # update state of inventory
                         current_inventory = update_inventory(
@@ -496,13 +499,17 @@ def decompose_subgoal(
     return subplan, current_inventory, action_items_used
 
 
-def get_subplans(observation: dict, return_items=False) -> tuple[list[list[str]], list]:
+def get_subplans(
+    observation: dict, return_items=False
+) -> tuple[list[list[str]], list, list[str]] | tuple[list[list[str]], list]:
     current_inventory = copy.deepcopy(observation["inventory"])
     plan = get_plan(observation)
     # get action
     if plan is None or len(plan) == 0:
+        if return_items:
+            return [[str(StopAction())]], [], []
         return [[str(StopAction())]], []
-    # plan_recipe, new_inventory = plan[0]
+
     subplans = []
     action_items_used = []
 

{plancraft-0.4.5.dist-info → plancraft-0.4.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: plancraft
-Version: 0.4.5
+Version: 0.4.7
 Summary: Plancraft: an evaluation dataset for planning with LLM agents
 License: MIT License
         

{plancraft-0.4.5.dist-info → plancraft-0.4.7.dist-info}/RECORD

@@ -11,13 +11,14 @@ plancraft/data/test.small.easy.json,sha256=5NZEJ2PqIgmHQecJOIVQyM1D6GFKyJq7GVmgR
 plancraft/data/test.small.json,sha256=eULAG1rdolRMXPrecV-7YoDIheKGyIT5MVpWdISV0wg,270089
 plancraft/data/train.json,sha256=asZIFnkBdgupPKRXacM4J0Ngt21B2BrMT6oPgFA96HI,2697710
 plancraft/data/val.json,sha256=IToAiaqUNQi_xhX1bzmInuskLaT7C2ryQjP-CZkzL24,1304403
+plancraft/data/val.repeated.json,sha256=gLeCg94oB0B_e8QyH8-FBygcJctGA4bkXYldoOHtSfQ,1726468
 plancraft/data/val.small.easy.json,sha256=9zEmqepjXG2NIp88xnFqOCkwsUsku3HEwHoQGxgTr6U,190252
 plancraft/data/val.small.json,sha256=76E9EFaljDQyAokg97e-IblvcOe6KbrdKkXvRxhhkgo,237653
 plancraft/environment/__init__.py,sha256=XFsFny4lH195AwAmL-WeCaF9ZCMgc7IgXIwhQ8FTdgE,505
 plancraft/environment/actions.py,sha256=Pub21caxM5iZ9IaX-ny1-xxr_peJIwwV_QAx3BVSry0,11551
 plancraft/environment/env.py,sha256=B7VpIMcQKITyDmkHgBoR4KbmxmM1b4A6Y-1_b90EMXo,16428
 plancraft/environment/items.py,sha256=Z9rhSyVDEoHF1pxRvhyiT94tyQJaWHi3wUHVcamz82o,221
-plancraft/environment/planner.py,sha256=-ZN6MSQxMTNfexa6ixzzFKK-qBSB4jbxOJLtknnHju8,20437
+plancraft/environment/planner.py,sha256=b8p4tH-EyC8v9Rc2wBQr9YJ8AS1dmG0LUBOsoba8Qd8,20695
 plancraft/environment/prompts.py,sha256=NU9YHAz3id-IgaukQvEi5uLlpEstpE5_Hccvvq1At2Y,6950
 plancraft/environment/recipes.py,sha256=0vwzOU86eZmGN2EpZVSIvzxpx0AOBWNPxTtAOFBN2A0,19570
 plancraft/environment/sampler.py,sha256=BworSMWQ-TLbV9068tkNOdo4ZLP-UDox6Laeb4Weu9k,7723
@@ -1923,7 +1924,7 @@ plancraft/models/generators.py,sha256=7COMLjjx_HbTWJqINNLqqExQv7gLikfLTViacAdSt5
 plancraft/models/oracle.py,sha256=f-0KWlBuHy6wcxmDsxM3MQ_QwfBstzfbA26mlk1MgLA,1657
 plancraft/models/utils.py,sha256=xgkP5jqCeFfkKe3Xd4ZYfTqiEJ-dA-qgFAC-J35ub3E,4029
 plancraft/train/dataset.py,sha256=oFqEd4LG9oEQ-71teh0Wf7-jJbtybT2ZibfM2bBdBkM,5474
-plancraft-0.4.5.dist-info/METADATA,sha256=60Kf6bWIVm32lmbZdgsYr1qcg8_KoycH-TxL5fw0Wg8,12391
-plancraft-0.4.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-plancraft-0.4.5.dist-info/licenses/LICENSE,sha256=YGR8ehDB4t-T-lOQKMfKNR-2zsOU7E3E5NA8t25HKE0,1070
-plancraft-0.4.5.dist-info/RECORD,,
+plancraft-0.4.7.dist-info/METADATA,sha256=hSaf9GuTMHGsBRW8wE4svMMIpfmL7yUVCtEFQKgwJw0,12391
+plancraft-0.4.7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+plancraft-0.4.7.dist-info/licenses/LICENSE,sha256=YGR8ehDB4t-T-lOQKMfKNR-2zsOU7E3E5NA8t25HKE0,1070
+plancraft-0.4.7.dist-info/RECORD,,