graph-games-proto 0.3.2096__py3-none-any.whl → 0.3.2100__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)
@@ -1159,7 +1170,6 @@ class PrivateState(PClass):
 class PublicDeck(PClass):
     idx = field(type=int)
     uuid = field(type=str)
-    units = field(type=list)  # List[Card]
     facedown_stack_len = field(type=int)
     discard_len = field(type=int)
     faceup_spots = field(type=list)  # List[Card]
@@ -1167,7 +1177,6 @@ class PublicDeck(PClass):
         return {
             "idx": self.idx,
             "uuid": self.uuid,
-            "units": [unit.__todict__() for unit in self.units],
             "facedown_stack_len": self.facedown_stack_len,
             "discard_len": self.discard_len,
             "faceup_spots": self.faceup_spots,
@@ -1177,7 +1186,6 @@ class PublicDeck(PClass):
         return PublicDeck(
             idx=d["idx"],
             uuid=d["uuid"],
-            units=[Card.__fromdict__(card) for card in d["units"]],
             facedown_stack_len=d["facedown_stack_len"],
             discard_len=d["discard_len"],
             faceup_spots=d["faceup_spots"],
@@ -1679,7 +1687,6 @@ class Action2(PClass):
 class Deck(PClass):
     idx = field(type=int)
     uuid = field(type=str)
-    units = field(type=list)  # List[Card]
     facedown_stack = field(type=list)  # List[str]
     discard = field(type=list)  # List[str]
     faceup_spots = field(type=list)  # List[str]
@@ -1687,7 +1694,6 @@ class Deck(PClass):
         return {
             "idx": self.idx,
             "uuid": self.uuid,
-            "units": [unit.__todict__() for unit in self.units],
             "facedown_stack": self.facedown_stack,
             "discard": self.discard,
             "faceup_spots": self.faceup_spots,
@@ -1697,7 +1703,6 @@ class Deck(PClass):
         return Deck(
             idx=d["idx"],
             uuid=d["uuid"],
-            units=[], # TODO: Card.__fromdict__(card) for card in d["cards"]
             facedown_stack=d["facedown_stack"],
             discard=d["discard"],
             faceup_spots=d["faceup_spots"],
@@ -2091,7 +2096,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 +2149,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"]],
@@ -2226,7 +2231,6 @@ def init_state_kernel(game_config, **kwargs):
         deck_obj = Deck(
             uuid=dek.uuid,
             idx=dek.idx,
-            units=[],
             faceup_spots=[],
             discard=[],
             facedown_stack=[card.uuid for card in cards],
@@ -4623,40 +4627,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)
 
@@ -4744,7 +4751,6 @@ def getpublicdeck(s, d):
     return PublicDeck(
         idx=d.idx,
         uuid=d.uuid,
-        units=d.units,
         facedown_stack_len=len(d.facedown_stack),
         discard_len=len(d.discard),
         faceup_spots=d.faceup_spots,
@@ -5564,10 +5570,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 +5588,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 +5602,7 @@ def dynamics(s, a):
     return next_s, rewards
 
 
+@dispatch(PlayerState)
 def alpha0(ps):
     td = 3
     legal_actions = ps.legal_actions_3
@@ -5600,7 +5610,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 +5620,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.2100.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: graph_games_proto
-Version: 0.3.2096
+Version: 0.3.2100
 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.2100.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=q656HgMAmHaIlbYI1lBYZLLTvRfa7X9tfCUoC-oQINI,199495
 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.2100.dist-info/METADATA,sha256=rpyhU4WzhqGJrINJ3gDW4851KSujWfNJX3sF5ZwJnBo,188
+graph_games_proto-0.3.2100.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+graph_games_proto-0.3.2100.dist-info/top_level.txt,sha256=-4QSrBMf_MM4BGsr2QXBpqDx8c8k_OPnzGyFjqjakes,18
+graph_games_proto-0.3.2100.dist-info/RECORD,,