valanga 0.1.6__tar.gz → 0.1.7__tar.gz

{valanga-0.1.6 → valanga-0.1.7}/PKG-INFO

@@ -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 → valanga-0.1.7}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "valanga"
-version = "0.1.6"
+version = "0.1.7"
 description = "Shared types and utilities for evaluation"
 requires-python = ">=3.13"
 dependencies = []

{valanga-0.1.6 → valanga-0.1.7}/src/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 → valanga-0.1.7}/src/valanga.egg-info/PKG-INFO

@@ -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