update model

Author: Dieg0Code

.github/workflows/ci-api.yml

@@ -48,6 +48,7 @@ jobs:
     timeout-minutes: 45
     env:
       PYTHONUNBUFFERED: "1"
+      UV_PYTHON: "3.11"
 
     steps:
       - name: Checkout
@@ -72,7 +73,13 @@ jobs:
           ln -sf ../bin/python .venv/Scripts/python.exe
 
       - name: Python file length policy
-        run: uv run python scripts/check_python_max_lines.py --max-lines 500 --path src --path tests --path scripts --path train.py
+        run: |
+          uv run python scripts/check_python_max_lines.py \
+            --max-lines 500 \
+            --path src/inference \
+            --path src/game \
+            --path src/data \
+            --path scripts
 
       - name: Ruff (API scope)
         run: uv run ruff check src tests scripts

.github/workflows/ci-train.yml

@@ -46,6 +46,7 @@ jobs:
     timeout-minutes: 35
     env:
       PYTHONUNBUFFERED: "1"
+      UV_PYTHON: "3.11"
 
     steps:
       - name: Checkout
@@ -80,8 +81,11 @@ jobs:
             --path src/game \
             --path src/data \
             --path tests/test_mcts_numerics.py \
+            --path tests/test_training_bootstrap.py \
+            --path tests/test_training_checkpointing.py \
+            --path tests/test_training_curriculum.py \
+            --path tests/test_training_monitor.py \
             --path tests/test_training_step_numerics.py \
-            --path tests/test_training_*.py \
             --path scripts/export_model_onnx.py \
             --path scripts/check_onnx_parity.py
 

Dockerfile

@@ -10,7 +10,7 @@ COPY web/ ./
 RUN npm run build -- --base=/web/static/
 
 
-FROM python:3.10-slim AS py-builder
+FROM python:3.11-slim AS py-builder
 
 ENV PYTHONDONTWRITEBYTECODE=1 \
     PYTHONUNBUFFERED=1 \
@@ -29,7 +29,7 @@ COPY pyproject.toml uv.lock README.md ./
 RUN uv sync --frozen --no-dev --group api
 
 
-FROM python:3.10-slim AS runtime
+FROM python:3.11-slim AS runtime
 
 ENV PYTHONDONTWRITEBYTECODE=1 \
     PYTHONUNBUFFERED=1 \

Dockerfile.train

@@ -1,6 +1,6 @@
 # syntax=docker/dockerfile:1.7
 
-FROM python:3.10-slim AS train-runtime
+FROM python:3.11-slim AS train-runtime
 
 ENV PYTHONDONTWRITEBYTECODE=1 \
     PYTHONUNBUFFERED=1 \

infra/runpod-train/__main__.py

@@ -11,7 +11,7 @@
 if str(SRC) not in sys.path:
     sys.path.insert(0, str(SRC))
 
-from training.runpod_infra import build_pod_env, build_train_start_command
+from training.runpod_infra import build_pod_env, build_train_start_command  # noqa: E402
 
 cfg = pulumi.Config()
 

main.py

@@ -1,4 +1,4 @@
-def main():
+def main() -> None:
     print("Hello from ataxx-zero!")
 
 

pyproject.toml

@@ -23,6 +23,7 @@ where = ["src"]
 [tool.ruff]
 line-length = 88
 target-version = "py310"
+extend-exclude = ["*.ipynb"]
 
 [tool.repo_policy]
 max_python_file_lines = 500

pyrefly.toml

@@ -1,13 +1,20 @@
 project-includes = [
     "src/**/*.py",
-    "main.py",
+    "tests/**/*.py",
+    "scripts/**/*.py",
+    "train.py",
+    "train_improved.py",
 ]
 
 project-excludes = [
     ".venv/**",
     "**/__pycache__",
     "**/.mypy_cache",
     "**/.pytest_cache",
+    "**/*.ipynb",
+    "infra/**",
+    "web/**",
+    ".github/**",
 ]
 
 python-interpreter-path = ".venv/Scripts/python.exe"

scripts/check_onnx_parity.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import argparse
+import importlib
 import sys
 from pathlib import Path
 from typing import TYPE_CHECKING, Any
@@ -62,7 +63,7 @@ def main() -> None:
     _ensure_src_on_path()
 
     try:
-        import onnxruntime as ort
+        ort = importlib.import_module("onnxruntime")
     except ImportError as exc:
         raise RuntimeError(
             "onnxruntime is required for parity checks. Install with `uv add --group dev onnxruntime`."

scripts/play_pygame.py

@@ -48,13 +48,11 @@
     from game.board import AtaxxBoard
     from game.types import Move
     from model.system import AtaxxZero
+    from ui.arena.effects import Particle
 
 PLAYER_1 = 1
 PLAYER_2 = -1
 
-Particle = dict[str, float | tuple[int, int, int]]
-
-
 def _ensure_src_on_path() -> None:
     if str(_SRC) not in sys.path:
         sys.path.insert(0, str(_SRC))
@@ -621,10 +619,11 @@ def main() -> None:
 
         if ai_turn and pending_apply_at is None:
             player = board.current_player
-            if ai_ready_at[player] is None:
+            ready_at = ai_ready_at[player]
+            if ready_at is None:
                 ai_ready_at[player] = now_ms + _ai_delay_ms(board, turn_agent, args.mcts_sims, rng)
                 status = f"{turn_agent} thinking..."
-            elif now_ms >= int(ai_ready_at[player]):
+            elif now_ms >= ready_at:
                 move, move_text = _pick_ai_move(
                     board=board,
                     agent=turn_agent,