graph-games-proto 0.3.1952__py3-none-any.whl → 0.3.1953__py3-none-any.whl

graph_games_proto/fns.py

@@ -1947,6 +1947,7 @@ class StateKernel(PClass):
     edgeuuid2idx = field(type=dict)  # Dict[str, int]
     carduuid2card = field(type=dict)  # Dict[str, Card]
     edgetuple2uuid = field(type=dict)  # Dict[Tuple[int, int], str]
+    nodeuuid2idx = field(type=dict)  # Dict[str, int]
     def __todict__(self):
         return {
             "rng": rng2json(self.rng),
@@ -1963,6 +1964,7 @@ class StateKernel(PClass):
             "edgeuuid2idx": self.edgeuuid2idx,
             "carduuid2card": {k: v.__todict__() for k, v in self.carduuid2card.items()},
             "edgetuple2uuid": [{"k": list(k), "v": v} for k, v in self.edgetuple2uuid.items()],
+            "nodeuuid2idx": self.nodeuuid2idx,
         }
     @staticmethod
     def __fromdict__(d):
@@ -1981,6 +1983,7 @@ class StateKernel(PClass):
             edgeuuid2idx=d["edgeuuid2idx"],
             carduuid2card={k: Card.__fromdict(v) for k, v in d["carduuid2card"].items()},
             edgetuple2uuid={tuple(item["k"]): item["v"] for item in d["edgetuple2uuid"]},
+            nodeuuid2idx=d["nodeuuid2idx"],
         )
 
 
@@ -2032,6 +2035,7 @@ def init_state_kernel(**kwargs):
         carduuid2card=carduuid2card,
         pieceuuid2piece=pieceuuid2piece,
         edgetuple2uuid=edgetuple2uuid,
+        nodeuuid2idx=nodeuuid2idx,
     )
 
 
@@ -2049,7 +2053,6 @@ class State(PClass):
     player_graphs = field(type=list)  # List[PlayerGraph]
     goals = field(type=list)  # List[Goal]
     nodes = field(type=list)  # List[Node]
-    nodeuuid2idx = field(type=dict)  # Dict[str, int]
     edges = field(type=list)  # List[BiEdge]
     regions = field(type=list)  # List[Region]
     legal_actions_3 = field(type=list)  # List[LegalAction]
@@ -2076,7 +2079,6 @@ class State(PClass):
             "player_graphs": [x.__todict__() for x in self.player_graphs],
             "goals": [goal.__todict__() for goal in self.goals],
             "nodes": [node.__todict__() for node in self.nodes],
-            "nodeuuid2idx": self.nodeuuid2idx,
             "edges": [edge.__todict__() for edge in self.edges],
             "regions": [region.__todict__() for region in self.regions],
             "legal_actions_3": [x.__todict__() for x in self.legal_actions_3],
@@ -2105,7 +2107,6 @@ class State(PClass):
             player_graphs=[PlayerGraph.__fromdict__(x) for x in d["player_graphs"]],
             goals=[FrozenGoal.__fromdict__(goal) for goal in d["goals"]],
             nodes=[Node.__fromdict__(n) for n in d["nodes"]],
-            nodeuuid2idx=d["nodeuuid2idx"],
             edges=[BiEdge.__fromdict__(e) for e in d["edges"]],
             regions=[Region.__fromdict__(r) for r in d["regions"]],
             legal_actions_3=[LegalAction.__fromdict__(x) for x in d["legal_actions_3"]],
@@ -2661,7 +2662,6 @@ def getinitialstate(game_config):
         ],
         goals = board_config.goals,
         nodes = nodes,
-        nodeuuid2idx = nodeuuid2idx,
         edges = edges,
         regions = get_regions(board_config),
         legal_actions_3=[],
@@ -4468,19 +4468,10 @@ def get_goal_completions(s, player_idx):
     ]
 
 
+@dispatch(State, FrozenGoal, int)
 def is_goal_complete(s, goal, player_idx):
     graph = s.player_graphs[player_idx].neighbors
-    nodeuuid2idx = {node.uuid: node.idx for node in s.nodes}
-    node_idxs = [nodeuuid2idx[node_uuid] for node_uuid in goal.node_uuids]
-    if goal.uuid == "d9ef7d75-63d4-4ceb-adce-f889db3c75f0":
-        print("goal.uuid: ", goal.uuid)
-        print("goal.node_uuids: ", goal.node_uuids)
-        print(f"player: {player_idx} goal: ", goal.uuid)
-        print("graph: ", graph)
-        print("node_idxs: ", node_idxs)
-        print("are_nodes_connected(graph, node_idxs): ", are_nodes_connected(graph, node_idxs))
-        for node in s.nodes:
-            print(f"node: {node}")
+    node_idxs = [s.kernel.nodeuuid2idx[node_uuid] for node_uuid in goal.node_uuids]
     return are_nodes_connected(
         graph,
         node_idxs,

{graph_games_proto-0.3.1952.dist-info → graph_games_proto-0.3.1953.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: graph_games_proto
-Version: 0.3.1952
+Version: 0.3.1953
 Requires-Dist: multipledispatch==1.0.0
 Requires-Dist: pyrsistent==0.20.0
 Requires-Dist: numpy==2.2.4

{graph_games_proto-0.3.1952.dist-info → graph_games_proto-0.3.1953.dist-info}/RECORD

@@ -1,9 +1,9 @@
 graph_games_proto/__init__.py,sha256=_EVQR-51XehfH45XZlba1WPdx3omS3Gm1nTwrgGyn2Q,667
 graph_games_proto/all_types.py,sha256=IpbwftEcHS5Ewz-saFNk0lO9FvcbuHG36odRTayCXUk,54911
-graph_games_proto/fns.py,sha256=k9npdHPzXr009dTSA_iJeTC7GiZj8TYohJlLyp1wULg,182582
+graph_games_proto/fns.py,sha256=Y880m_lmU6FCvxx4zowDu9Lrgcco4pOdDYYYQXzrQNY,182121
 graph_games_proto/main.py,sha256=fj2U7KcwrpZtuUhjOX5yVxY18LZvvsxDFYZ_S5mxe04,145
 graph_games_proto/state.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-graph_games_proto-0.3.1952.dist-info/METADATA,sha256=5BHTXHDTovOFX5UTg17YNTXEHi2_Q1vTPpfCl6vJkwA,188
-graph_games_proto-0.3.1952.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-graph_games_proto-0.3.1952.dist-info/top_level.txt,sha256=-4QSrBMf_MM4BGsr2QXBpqDx8c8k_OPnzGyFjqjakes,18
-graph_games_proto-0.3.1952.dist-info/RECORD,,
+graph_games_proto-0.3.1953.dist-info/METADATA,sha256=ZAxOkRoFWLDz0NzLq1q15j0-FWw_gYTHWiqVvOH_Is8,188
+graph_games_proto-0.3.1953.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+graph_games_proto-0.3.1953.dist-info/top_level.txt,sha256=-4QSrBMf_MM4BGsr2QXBpqDx8c8k_OPnzGyFjqjakes,18
+graph_games_proto-0.3.1953.dist-info/RECORD,,