valanga 0.1.6__py3-none-any.whl → 0.1.7__py3-none-any.whl

valanga/representation_factory.py

@@ -3,23 +3,35 @@ Factory for creating content representations from game states and state modifica
 """
 
 from dataclasses import dataclass
-from typing import Callable, TypeVar
+from typing import Protocol
 
-from .game import State, StateModifications
+from .game import State
 from .represention_for_evaluation import ContentRepresentation
 
-StateT = TypeVar("StateT", bound=State)
-EvalIn = TypeVar("EvalIn")
 
-CreateFromState = Callable[[StateT], ContentRepresentation[StateT, EvalIn]]
-CreateFromStateAndMods = Callable[
-    [StateT, StateModifications, ContentRepresentation[StateT, EvalIn]],
-    ContentRepresentation[StateT, EvalIn],
-]
+class CreateFromState[StateT: State, EvalIn](Protocol):
+    """
+    Protocol for creating a state representation from a state.
+    """
+
+    def __call__(self, state: StateT) -> ContentRepresentation[StateT, EvalIn]: ...
+
+
+class CreateFromStateAndModifications[StateT: State, EvalIn, StateModT](Protocol):
+    """
+    Protocol for creating a state representation from a state and modifications.
+    """
+
+    def __call__(
+        self,
+        state: StateT,
+        state_modifications: StateModT,
+        previous_state_representation: ContentRepresentation[StateT, EvalIn],
+    ) -> ContentRepresentation[StateT, EvalIn]: ...
 
 
 @dataclass
-class RepresentationFactory[StateT: State, EvalIn]:
+class RepresentationFactory[StateT: State, EvalIn, StateModT]:
     """Factory for creating content representations from states and state modifications.
     Attributes:
         create_from_state: Function to create a content representation from a state.
@@ -27,13 +39,15 @@ class RepresentationFactory[StateT: State, EvalIn]:
     """
 
     create_from_state: CreateFromState[StateT, EvalIn]
-    create_from_state_and_modifications: CreateFromStateAndMods[StateT, EvalIn]
+    create_from_state_and_modifications: CreateFromStateAndModifications[
+        StateT, EvalIn, StateModT
+    ]
 
     def create_from_transition(
         self,
         state: StateT,
         previous_state_representation: ContentRepresentation[StateT, EvalIn] | None,
-        modifications: StateModifications | None,
+        modifications: StateModT | None,
     ) -> ContentRepresentation[StateT, EvalIn]:
         """Creates a content representation from a state transition.
         Args:

{valanga-0.1.6.dist-info → valanga-0.1.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: valanga
-Version: 0.1.6
+Version: 0.1.7
 Summary: Shared types and utilities for evaluation
 Author-email: Victor Gabillon <victorgabillon@gmail.com>
 License:                     GNU GENERAL PUBLIC LICENSE

{valanga-0.1.6.dist-info → valanga-0.1.7.dist-info}/RECORD

@@ -6,10 +6,10 @@ valanga/over_event.py,sha256=jUqN3CXq3e55TvDZatZKFlEOzJfho7M1Zr7VjZUhP6M,7908
 valanga/policy.py,sha256=TkqcOSePRhHqtuoMbCKe5qZbizvF5H9uG1_oEcIZsuA,1229
 valanga/progress_messsage.py,sha256=3LnOXSgyU84UkwcpsakcbPt_d3el2ojRSUClAwoPyEk,361
 valanga/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-valanga/representation_factory.py,sha256=w1RKwuDec-imp5NmB3rg-e5iiuEScbHCX8tnBeXeVgQ,2117
+valanga/representation_factory.py,sha256=LAIfLCBKTMVPDCZbcRtMQs_VIJwIHeaEH86BWV8PgTc,2456
 valanga/represention_for_evaluation.py,sha256=9yCCOuN-h9FnoLJrffKGQ-oXiiXzZT3rP4e5wwMYfys,1135
-valanga-0.1.6.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-valanga-0.1.6.dist-info/METADATA,sha256=cMwv58HEFZHTEmRujAzCRcnU0uL24voleiQV7OVCZDE,44645
-valanga-0.1.6.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
-valanga-0.1.6.dist-info/top_level.txt,sha256=cKFLhxKDTqygHnc0xc_L8jDx6oNu5hSHqZ4UExU4xPo,8
-valanga-0.1.6.dist-info/RECORD,,
+valanga-0.1.7.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+valanga-0.1.7.dist-info/METADATA,sha256=4qCW9MTi80M1_3RJSi2hNiQlqV70UbzaZ0_YbdRvjTU,44645
+valanga-0.1.7.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
+valanga-0.1.7.dist-info/top_level.txt,sha256=cKFLhxKDTqygHnc0xc_L8jDx6oNu5hSHqZ4UExU4xPo,8
+valanga-0.1.7.dist-info/RECORD,,