valanga 0.1.2__tar.gz → 0.1.3__tar.gz

{valanga-0.1.2/src/valanga.egg-info → valanga-0.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: valanga
-Version: 0.1.2
+Version: 0.1.3
 Summary: Shared types and utilities for evalaution
 Requires-Python: >=3.13
 License-File: LICENSE
@@ -8,6 +8,12 @@ Provides-Extra: test
 Requires-Dist: pytest>=8.4.1; extra == "test"
 Requires-Dist: coverage; extra == "test"
 Requires-Dist: pytest-cov>=6.0.0; extra == "test"
+Provides-Extra: lint
+Requires-Dist: ruff>=0.14.10; extra == "lint"
+Requires-Dist: pylint>=4.0.4; extra == "lint"
+Provides-Extra: typecheck
+Requires-Dist: mypy>=1.19.1; extra == "typecheck"
+Requires-Dist: pyright[nodejs]>=1.1.408; extra == "typecheck"
 Provides-Extra: dev
 Requires-Dist: tox>=4.32.0; extra == "dev"
 Requires-Dist: types-PyYAML>=6.0.12.12; extra == "dev"
@@ -15,7 +21,7 @@ Requires-Dist: ruff>=0.14.10; extra == "dev"
 Requires-Dist: pylint>=4.0.4; extra == "dev"
 Requires-Dist: black>=25.12.0; extra == "dev"
 Requires-Dist: mypy>=1.19.1; extra == "dev"
-Requires-Dist: pyright>=1.1.407; extra == "dev"
+Requires-Dist: pyright[nodejs]>=1.1.408; extra == "dev"
 Requires-Dist: build; extra == "dev"
 Requires-Dist: twine; extra == "dev"
 Requires-Dist: sphinx; extra == "dev"

{valanga-0.1.2 → valanga-0.1.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "valanga"
-version = "0.1.2"
+version = "0.1.3"
 description = "Shared types and utilities for evalaution"
 requires-python = ">=3.13"
 dependencies = []
@@ -20,6 +20,16 @@ test = [
     "pytest-cov>=6.0.0"
 ]
 
+lint = [
+  "ruff>=0.14.10",
+  "pylint>=4.0.4",
+]
+
+typecheck = [
+  "mypy>=1.19.1",
+  "pyright[nodejs]>=1.1.408",
+]
+
 
 dev = [
     # Development dependencies (includes test)
@@ -33,7 +43,7 @@ dev = [
 
     # Type checking
     "mypy>=1.19.1",
-    "pyright>=1.1.407   ",
+    "pyright[nodejs]>=1.1.408   ",
 
     # Building & publishing
     "build",
@@ -150,4 +160,44 @@ reportMissingImports = true
 reportUnusedImport = true
 reportMissingTypeStubs = false
 reportUnknownMemberType = false
-reportPrivateImportUsage = false
+reportPrivateImportUsage = false
+
+
+#################### TOX ####################
+
+[tool.tox]
+min_version = "4.32.0"
+env_list = ["py313", "lint", "typecheck"]
+isolated_build = true
+
+[tool.tox.env_run_base]
+base_python = ["python3.13"]
+
+[tool.tox.env.py313]
+description = "Run tests"
+extras = ["test"]
+commands = [["pytest", "{posargs}"]]
+
+[tool.tox.env.lint]
+description = "ruff + pylint (fast, no package install)"
+package = "skip"
+deps = ["ruff>=0.14.10", "pylint>=4.0.4"]
+set_env = { PYTHONPATH = "{toxinidir}/src" }
+commands = [
+  ["python", "-m", "ruff", "check", "src", "tests"],
+  ["python", "-m", "ruff", "format", "--check", "src", "tests"],
+  ["python", "-m", "pylint", "src/valanga"],
+]
+
+[tool.tox.env.typecheck]
+description = "mypy + pyright (fast, no package install)"
+package = "skip"
+deps = ["mypy>=1.19.1", "pyright[nodejs]>=1.1.408"]
+set_env = { MYPYPATH = "{toxinidir}/src" }
+commands = [
+  ["python", "-m", "mypy", "--config-file", "pyproject.toml", "src/valanga"],
+  ["python", "-m", "pyright", "src/valanga"],
+]
+
+
+#################### END TOX ####################

{valanga-0.1.2 → valanga-0.1.3}/src/valanga/__init__.py

@@ -1,6 +1,6 @@
 """Common types and utilities shared by multiple libraries."""
 
-from .evaluations import BoardEvaluation, EvalItem, FloatyStateEvaluation, ForcedOutcome
+from .evaluations import EvalItem, FloatyStateEvaluation, ForcedOutcome, StateEvaluation
 from .game import (
     BLACK,
     WHITE,
@@ -22,7 +22,7 @@ from .represention_for_evaluation import ContentRepresentation
 __all__ = [
     "ForcedOutcome",
     "FloatyStateEvaluation",
-    "BoardEvaluation",
+    "StateEvaluation",
     "EvalItem",
     "OverEvent",
     "Color",

{valanga-0.1.2 → valanga-0.1.3}/src/valanga/evaluations.py

@@ -55,4 +55,4 @@ class FloatyStateEvaluation:
     value_white: float | None
 
 
-BoardEvaluation = FloatyStateEvaluation | ForcedOutcome
+StateEvaluation = FloatyStateEvaluation | ForcedOutcome

{valanga-0.1.2 → valanga-0.1.3}/src/valanga/game.py

@@ -6,6 +6,8 @@ from collections.abc import Hashable
 from enum import Enum
 from typing import Annotated, Iterator, Protocol, Self, Sequence, TypeVar
 
+type Seed = Annotated[int, "seed"]
+
 type StateTag = Annotated[Hashable, "A label or identifier for a state in a game"]
 
 type StateModifications = Annotated[

valanga-0.1.3/src/valanga/policy.py

@@ -0,0 +1,42 @@
+"""
+Policy-related classes and protocols for branch selection in game trees.
+"""
+
+from dataclasses import dataclass
+from typing import Mapping, Protocol
+
+from valanga.evaluations import StateEvaluation
+from valanga.game import BranchKey, Seed, State
+
+
+@dataclass(frozen=True, slots=True)
+class BranchPolicy:
+    """
+    Represents a probability distribution over branches.
+    """
+
+    probs: Mapping[BranchKey, float]  # should sum to ~1.0
+
+
+@dataclass(frozen=True, slots=True)
+class Recommendation:
+    """A recommendation for a specific branch in a tree node."""
+
+    recommended_key: BranchKey
+    evaluation: StateEvaluation | None = None
+    policy: BranchPolicy | None = None
+    branch_evals: Mapping[BranchKey, StateEvaluation] | None = None
+
+
+class BranchSelector(Protocol):
+    """Protocol for a branch selector."""
+
+    def recommend(self, state: State, seed: Seed) -> Recommendation:
+        """Given a state and a seed, recommends a branch to take.
+        Args:
+            state (State): The current state of the game.
+            seed (Seed): A seed for any randomness involved in the selection.
+        Returns:
+            Recommendation: The recommended branch to take.
+        """
+        ...

{valanga-0.1.2 → valanga-0.1.3/src/valanga.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: valanga
-Version: 0.1.2
+Version: 0.1.3
 Summary: Shared types and utilities for evalaution
 Requires-Python: >=3.13
 License-File: LICENSE
@@ -8,6 +8,12 @@ Provides-Extra: test
 Requires-Dist: pytest>=8.4.1; extra == "test"
 Requires-Dist: coverage; extra == "test"
 Requires-Dist: pytest-cov>=6.0.0; extra == "test"
+Provides-Extra: lint
+Requires-Dist: ruff>=0.14.10; extra == "lint"
+Requires-Dist: pylint>=4.0.4; extra == "lint"
+Provides-Extra: typecheck
+Requires-Dist: mypy>=1.19.1; extra == "typecheck"
+Requires-Dist: pyright[nodejs]>=1.1.408; extra == "typecheck"
 Provides-Extra: dev
 Requires-Dist: tox>=4.32.0; extra == "dev"
 Requires-Dist: types-PyYAML>=6.0.12.12; extra == "dev"
@@ -15,7 +21,7 @@ Requires-Dist: ruff>=0.14.10; extra == "dev"
 Requires-Dist: pylint>=4.0.4; extra == "dev"
 Requires-Dist: black>=25.12.0; extra == "dev"
 Requires-Dist: mypy>=1.19.1; extra == "dev"
-Requires-Dist: pyright>=1.1.407; extra == "dev"
+Requires-Dist: pyright[nodejs]>=1.1.408; extra == "dev"
 Requires-Dist: build; extra == "dev"
 Requires-Dist: twine; extra == "dev"
 Requires-Dist: sphinx; extra == "dev"

{valanga-0.1.2 → valanga-0.1.3}/src/valanga.egg-info/SOURCES.txt

@@ -6,6 +6,7 @@ src/valanga/evaluations.py
 src/valanga/evaluator_types.py
 src/valanga/game.py
 src/valanga/over_event.py
+src/valanga/policy.py
 src/valanga/progress_messsage.py
 src/valanga/py.typed
 src/valanga/representation_factory.py

{valanga-0.1.2 → valanga-0.1.3}/src/valanga.egg-info/requires.txt

@@ -6,7 +6,7 @@ ruff>=0.14.10
 pylint>=4.0.4
 black>=25.12.0
 mypy>=1.19.1
-pyright>=1.1.407
+pyright[nodejs]>=1.1.408
 build
 twine
 sphinx
@@ -14,7 +14,15 @@ sphinx-rtd-theme
 sphinx-autodoc-typehints
 pre-commit
 
+[lint]
+ruff>=0.14.10
+pylint>=4.0.4
+
 [test]
 pytest>=8.4.1
 coverage
 pytest-cov>=6.0.0
+
+[typecheck]
+mypy>=1.19.1
+pyright[nodejs]>=1.1.408