graph-games-proto 0.3.2095__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"]],
@@ -2551,6 +2562,7 @@ class PublicState(PClass):
     terminal = field(type=bool)
     def __todict__(self):
         return {
+            "game_config": self.game_config.__todict__(),
             "deadlines": [deadline.__todict__() if deadline else None for deadline in self.deadlines],
             "game_started_at": self.game_started_at,
             "allotted_times": [allotted_time.__todict__() if allotted_time else None for allotted_time in self.allotted_times], # List[AllottedTime|None]
@@ -2577,6 +2589,7 @@ class PublicState(PClass):
     @staticmethod
     def __fromdict__(d):
         return PublicState(
+            game_config=PublicGameConfig.__fromdict__(d["game_config"]),
             deadlines=[RemainingAllottedTime.__fromdict__(x) for x in d["deadlines"]],
             game_started_at=d["game_started_at"],
             allotted_times=[AllottedTime.__fromdict__(x) if x else None for x in d["allotted_times"]], # List[AllottedTime|None]
@@ -4557,7 +4570,7 @@ def get_deadline(s, max_allotted_time):
     since_action_idx = max_allotted_time.since_action_idx
     if since_action_idx == -1:
         allotted_since = datetime.strptime(
-            s.kernel.started_at, 
+            s.kernel.game_config.started_at, 
             "%Y-%m-%d %H:%M:%S"
         )
     else:
@@ -4621,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)
 
@@ -4713,7 +4729,7 @@ def getpublicstate(s):
     return PublicState(
         game_config=getpublicgameconfig(s.kernel.game_config),
         deadlines=get_deadlines(s),
-        game_started_at=s.kernel.started_at,
+        game_started_at=s.kernel.game_config.started_at,
         allotted_times=get_max_allotted_times(s),
         all_pieces=list(s.kernel.pieceuuid2piece.values()),
         to_play_2=getpublictoplay(s),
@@ -5562,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):
@@ -5579,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)
@@ -5591,6 +5610,7 @@ def dynamics(s, a):
     return next_s, rewards
 
 
+@dispatch(PlayerState)
 def alpha0(ps):
     td = 3
     legal_actions = ps.legal_actions_3
@@ -5598,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]
 
 
@@ -5608,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.2095.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.2095
+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.2095.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=aq2Vxkuz1J4LGAPlO8OZtW1Tns81gL68UICMfsZzeHY,199556
+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.2095.dist-info/METADATA,sha256=3B4hZJi4LhHbTKGGFf-y7t0nTc0sQ-w_1rPTCZ0-jyA,188
-graph_games_proto-0.3.2095.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-graph_games_proto-0.3.2095.dist-info/top_level.txt,sha256=-4QSrBMf_MM4BGsr2QXBpqDx8c8k_OPnzGyFjqjakes,18
-graph_games_proto-0.3.2095.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,,