plancraft 0.3.24__py3-none-any.whl → 0.3.26__py3-none-any.whl

plancraft/environment/actions.py

@@ -22,6 +22,10 @@ def convert_to_slot_index(slot: str) -> int:
     if slot in grid_map:
         return grid_map[slot]
     else:
+        # check if it is an inventory slot
+        if len(slot) < 4 or len(slot) > 5 or slot[1] != "I" or slot[2] == "0":
+            raise ValueError("Invalid slot index")
+        # convert I1 to 10, I2 to 11, I3 to 12, ..., I36 to 45
         return int(slot[2:-1]) + 9
 
 
@@ -40,8 +44,9 @@ def convert_from_slot_index(slot_index: int) -> str:
     }
     if slot_index < 10:
         return grid_map[slot_index]
-    else:
+    elif slot_index < 46:
         return f"[I{slot_index-9}]"
+    raise ValueError("Invalid slot index")
 
 
 class ActionHandlerBase(abc.ABC):
@@ -123,9 +128,13 @@ class MoveAction(BaseModel):
         if self.slot_from == self.slot_to:
             raise AttributeError("[Source] and [Target] must be different")
         if self.slot_from < 0 or self.slot_from > 45:
-            raise AttributeError("[Source] must be between 0 and 45")
+            raise AttributeError(
+                "[Source] must be [0] or [A1] to [C3] or [I1] to [I36]"
+            )
+        if self.slot_to == 0:
+            raise AttributeError("You cannot move items into [0]")
         if self.slot_to < 1 or self.slot_to > 45:
-            raise AttributeError("[Target] must be between 1 and 45")
+            raise AttributeError("[Target] must be [A1] to [C3] or [I1] to [I36]")
         if self.quantity < 1 or self.quantity > 64:
             raise AttributeError("quantity must be between 1 and 64")
         return self
@@ -172,9 +181,13 @@ class SmeltAction(BaseModel):
         if self.slot_from == self.slot_to:
             raise AttributeError("[Source] and [Target] must be different")
         if self.slot_from < 0 or self.slot_from > 45:
-            raise AttributeError("[Source] must be between 0 and 45")
+            raise AttributeError(
+                "[Source] must be [0] or [A1] to [C3] or [I1] to [I36]"
+            )
+        if self.slot_to == 0:
+            raise AttributeError("You cannot smelt items into [0]")
         if self.slot_to < 1 or self.slot_to > 45:
-            raise AttributeError("[Target] must be between 1 and 45")
+            raise AttributeError("[Target] must be [A1] to [C3] or [I1] to [I36]")
         if self.quantity < 1 or self.quantity > 64:
             raise AttributeError("quantity must be between 1 and 64")
         return self

{plancraft-0.3.24.dist-info → plancraft-0.3.26.dist-info}/METADATA

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

{plancraft-0.3.24.dist-info → plancraft-0.3.26.dist-info}/RECORD

@@ -11,7 +11,7 @@ plancraft/data/val.json,sha256=IToAiaqUNQi_xhX1bzmInuskLaT7C2ryQjP-CZkzL24,13044
 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=Ve3SHwg0o2NcMBVM8O-IRjsvcdO2dH8kqvweWm6euo8,10559
+plancraft/environment/actions.py,sha256=9GESxqx4z0aTwEmI6H-8MJMaqnFTAgsfck7GOHUQEBg,11175
 plancraft/environment/env.py,sha256=A4532st7JFBYBF_Nh0CEEi3ZTLJAeaB3t9PAIVSemj0,16390
 plancraft/environment/items.py,sha256=Z9rhSyVDEoHF1pxRvhyiT94tyQJaWHi3wUHVcamz82o,221
 plancraft/environment/planner.py,sha256=uIOJjIoyT_4pxeWeTKb8BkLJyKZG0-AMoEOkZs6Ua9A,19340
@@ -1920,7 +1920,7 @@ plancraft/models/generators.py,sha256=F76_iPiqxUjDIrQwF58tzM0bLM91OkZJ0sBqBuki5w
 plancraft/models/oracle.py,sha256=f-0KWlBuHy6wcxmDsxM3MQ_QwfBstzfbA26mlk1MgLA,1657
 plancraft/models/utils.py,sha256=E-sZohvolWgGbpHQKgAgkgIfUJoVnT5pMt6JP8xLHKg,4034
 plancraft/train/dataset.py,sha256=oFqEd4LG9oEQ-71teh0Wf7-jJbtybT2ZibfM2bBdBkM,5474
-plancraft-0.3.24.dist-info/METADATA,sha256=yCmPq3zXC2cEu5E9c1MOT1nsrfGcI_clOMYIV4GQoR4,11148
-plancraft-0.3.24.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-plancraft-0.3.24.dist-info/licenses/LICENSE,sha256=YGR8ehDB4t-T-lOQKMfKNR-2zsOU7E3E5NA8t25HKE0,1070
-plancraft-0.3.24.dist-info/RECORD,,
+plancraft-0.3.26.dist-info/METADATA,sha256=2s3S8xZMyP8v1T0y2qHl5OHIN18sMn0UXXA8iJxUQq8,11148
+plancraft-0.3.26.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+plancraft-0.3.26.dist-info/licenses/LICENSE,sha256=YGR8ehDB4t-T-lOQKMfKNR-2zsOU7E3E5NA8t25HKE0,1070
+plancraft-0.3.26.dist-info/RECORD,,