graph-games-proto 0.3.1898__py3-none-any.whl → 0.3.1900__py3-none-any.whl

graph_games_proto/fns.py

@@ -4078,10 +4078,26 @@ class StateKernal(PClass):
     history = field(type=list)  # List[PublicAction]
     player_scores = field(type=list)  # List[PlayerScore]
     game_config = field(type=GameConfig)
+    uuid2edge = field(type=dict)  # Dict[str, BiEdge]
+    idx2path = field(type=list)  # List[Path2]
+    uuid2segment = field(type=dict)  # Dict[str, Segment]
+    edgeuuid2idx = field(type=dict)  # Dict[str, int]
+
 
 def init_state_kernal(**kwargs):
+    edges = kwargs.get('edges', [])
+    uuid2edge = {}
+    idx2path = []
+    uuid2segment = {}
+    edgeuuid2idx = {edge.uuid: idx for idx, edge in enumerate(edges)}
+    for edge in edges:
+        uuid2edge[edge.uuid] = edge
+        for path in edge.paths:
+            idx2path.append(path)
+            for segment in path.segments:
+                uuid2segment[segment.uuid] = segment
     return StateKernal(
-        edges=kwargs.get('edges'),
+        edges=edges,
         decks=kwargs.get('decks'),
         piles=kwargs.get('piles'),
         players=kwargs.get('players'),
@@ -4089,10 +4105,13 @@ def init_state_kernal(**kwargs):
         history=kwargs.get('history'),
         player_scores=kwargs.get('player_scores'),
         game_config=kwargs.get('game_config'),
+        uuid2edge=uuid2edge,
+        idx2path=idx2path,
+        uuid2segment=uuid2segment,
+        edgeuuid2idx=edgeuuid2idx,
     )
 
 
-
 def getnextstate2(s, a, log=False):
     is_legal, reason = isactionlegal2(s, a)
     if not is_legal:

{graph_games_proto-0.3.1898.dist-info → graph_games_proto-0.3.1900.dist-info}/METADATA

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

{graph_games_proto-0.3.1898.dist-info → graph_games_proto-0.3.1900.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=DJTM-VwAx-5vcXtBLGgdczV80jCGkP3_Jake_GMMw0U,187280
+graph_games_proto/fns.py,sha256=UaOPciy5NGI80NFPQ7UkCAh76MTngEOQfwdd9v-Zgs4,187990
 graph_games_proto/main.py,sha256=fj2U7KcwrpZtuUhjOX5yVxY18LZvvsxDFYZ_S5mxe04,145
 graph_games_proto/state.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-graph_games_proto-0.3.1898.dist-info/METADATA,sha256=j3adFTTmlHyIBV75pT5bO9GEEEN3LPvNQhgVIfPf57I,188
-graph_games_proto-0.3.1898.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-graph_games_proto-0.3.1898.dist-info/top_level.txt,sha256=-4QSrBMf_MM4BGsr2QXBpqDx8c8k_OPnzGyFjqjakes,18
-graph_games_proto-0.3.1898.dist-info/RECORD,,
+graph_games_proto-0.3.1900.dist-info/METADATA,sha256=AHrt9_-p3mdMs6pDQ0nrl_Vzly7iMgnU7IeqofXelbk,188
+graph_games_proto-0.3.1900.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+graph_games_proto-0.3.1900.dist-info/top_level.txt,sha256=-4QSrBMf_MM4BGsr2QXBpqDx8c8k_OPnzGyFjqjakes,18
+graph_games_proto-0.3.1900.dist-info/RECORD,,