graph-games-proto 0.3.2096__py3-none-any.whl → 0.3.2099__py3-none-any.whl

graph_games_proto/fns.py

@@ -1063,7 +1063,18 @@ class Player(PClass):
             cards=d["cards"],
             discard_tray=d["discard_tray"]
         )
-    
+
+
+@dispatch(PublicPlayer)
+def imagine_player(public_player) -> Player:
+    return Player(
+        idx=public_player.idx,
+        pieces=public_player.pieces,
+        cards=[],
+        discard_tray=[],
+    )
+
+
 class GoalCompletion(PClass):
     goal_uuid = field(type=str)
     complete = field(type=bool, initial=False)
@@ -2091,7 +2102,7 @@ class StateKernel(PClass):
     piles = field(type=list)  # List[Pile]
     players = field(type=list)  # List[Player]
     player_idxs = field(type=list)  # List[int]
-    history = field(type=list)  # List[PublicAction]
+    history = field(type=list)  # List[Action2]
     pieceuuid2piece = field(type=dict)  # Dict[str, Piece]
     edgeuuid2idx = field(type=dict)  # Dict[str, int]
     carduuid2card = field(type=dict)  # Dict[str, Card]
@@ -2144,7 +2155,7 @@ class StateKernel(PClass):
             piles=[Pile.__fromdict__(pile) for pile in d["piles"]],
             players=[Player.__fromdict__(player) for player in d["players"]],
             player_idxs=d["player_idxs"],
-            history=[PublicAction.__fromdict__(action) for action in d["history"]],
+            history=[Action2.__fromdict__(action) for action in d["history"]],
             uuid2edgerecord={k: BiEdgeRecord.__fromdict__(v) for k, v in d["uuid2edgerecord"].items()},
             uuid2segmentrecord={k: Segment2.__fromdict__(v) for k, v in d["uuid2segmentrecord"].items()},
             idx2pathrecord=[Path2.__fromdict(v) for v in d["idx2pathrecord"]],
@@ -4623,40 +4634,43 @@ def get_public_history(s):
     return [action.get_public(s) for action in s.kernel.history]
 
 
-def imagine_history(public_state, private_state):
+@dispatch(PlayerState)
+def imagine_history(ps) -> list[Action2]:
     pass
 
-def imagine_player_scores(public_state, private_state):
-    pass
 
-def imagine_legal_actions(public_state, private_state):
+@dispatch(PlayerState)
+def imagine_players(ps) -> list[Player]:
     pass
 
-def imagine_players(public_state, private_state):
-    pass
 
-def imagine_decks(public_state, private_state):
+@dispatch(PlayerState)
+def imagine_decks(ps) -> list[Deck]:
     pass
 
 
-def imagine_state(public_state, private_state):
-    public_game_config = public_state.game_config
+@dispatch(PlayerState)
+def imagine_state(ps):
+    public = ps.public
+    game_config = public.game_config
+
+    # candidates = get_candidates(ps)
 
     imagined_kernel = init_state_kernel(
         GameConfig(
             seed=random.randint(0, 2**31 - 1),
-            uuid=public_game_config.uuid,
-            num_players=public_game_config.num_players,
-            fig=public_game_config.fig,
-            started_at=public_game_config.started_at,
+            uuid=game_config.uuid,
+            num_players=game_config.num_players,
+            fig=game_config.fig,
+            started_at=game_config.started_at,
         ),
-        edges=public_state.edges,
-        nodes=public_state.nodes,
-        piles=public_state.piles,
-        player_idxs=public_state.player_idxs,
-        decks=imagine_decks(public_state, private_state),
-        players=imagine_players(public_state, private_state),
-        history=imagine_history(public_state, private_state),
+        edges=public.edges,
+        nodes=public.nodes,
+        piles=public.piles,
+        player_idxs=public.player_idxs,
+        decks=imagine_decks(ps),
+        players=imagine_players(ps),
+        history=imagine_history(ps),
     )
     return init_memoized_state(imagined_kernel)
 
@@ -5564,10 +5578,11 @@ def get_default_toplay(s):
     return None
 
 
+@dispatch(PlayerState)
 def get_intuited_best_actions(ps):
-    if not ps.legal_actions_3:
+    if not ps.private.legal_actions_3:
         return None
-    return ps.legal_actions_3[:8]
+    return ps.private.legal_actions_3[:8]
 
 
 def get_spread(q_values, p_idx):
@@ -5581,10 +5596,12 @@ def getvproxy0(ps):
     return 0
 
 
+@dispatch(PlayerState, Action2)
 def imagine_dynamics(ps, a):
     return dynamics(imagine_state(ps), a)
 
 
+@dispatch(State, Action2)
 def dynamics(s, a):
     scores = get_public_player_scores(s)
     next_s = getnextstate2(s, a)
@@ -5593,6 +5610,7 @@ def dynamics(s, a):
     return next_s, rewards
 
 
+@dispatch(PlayerState)
 def alpha0(ps):
     td = 3
     legal_actions = ps.legal_actions_3
@@ -5600,7 +5618,7 @@ def alpha0(ps):
         return None
     intuited = get_intuited_best_actions(ps)
     q_proxies = [getqproxy0(ps, a, td) for a in intuited]
-    max_spread_idx = get_max_spread_idx(q_proxies, ps.player.player_idx)
+    max_spread_idx = get_max_spread_idx(q_proxies, ps.private.player.player_idx)
     return intuited[max_spread_idx]
 
 
@@ -5610,6 +5628,7 @@ def get_max_spread_idx(q_proxies, p_idx):
     return max_spread_idx
 
 
+@dispatch(PlayerState, Action2, int)
 def getqproxy0(ps, a, td):
 
     def qproxybase():

{graph_games_proto-0.3.2096.dist-info → graph_games_proto-0.3.2099.dist-info}/METADATA

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

{graph_games_proto-0.3.2096.dist-info → graph_games_proto-0.3.2099.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=Ni2ywvhv9562Hi1_pQie5P4aEtqLORss3yxaF-b24L8,199711
+graph_games_proto/fns.py,sha256=c5Vj7Ye-LnNQBbOm7tQg0EWvcinATUSeDLWduCdrHaw,199901
 graph_games_proto/main.py,sha256=fj2U7KcwrpZtuUhjOX5yVxY18LZvvsxDFYZ_S5mxe04,145
 graph_games_proto/state.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-graph_games_proto-0.3.2096.dist-info/METADATA,sha256=RsItWXIDMFkMIN2peGo0mTJo6zq6UpjN64X5kChH3qE,188
-graph_games_proto-0.3.2096.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-graph_games_proto-0.3.2096.dist-info/top_level.txt,sha256=-4QSrBMf_MM4BGsr2QXBpqDx8c8k_OPnzGyFjqjakes,18
-graph_games_proto-0.3.2096.dist-info/RECORD,,
+graph_games_proto-0.3.2099.dist-info/METADATA,sha256=tsn8lVkH1Vnv04mvMZhdfh8vRmWeDEW65g-VkZ5GXRQ,188
+graph_games_proto-0.3.2099.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+graph_games_proto-0.3.2099.dist-info/top_level.txt,sha256=-4QSrBMf_MM4BGsr2QXBpqDx8c8k_OPnzGyFjqjakes,18
+graph_games_proto-0.3.2099.dist-info/RECORD,,