pint units (GEP 10) · infra 2/3: currency & boundary

.github/workflows/main.yml

@@ -92,10 +92,10 @@ jobs:
           cache: true
           cache-write: ${{ github.event_name == 'push' && github.ref_name == 'main' }}
           frozen: true
-          environments: type-checking type-checking-jax
-      - name: Run ty
-        run: pixi run --locked ty
+          environments: py314 py314-jax
+      - name: Run ty (numpy backend)
+        run: pixi run --locked -e py314 prek run ty --all-files
         shell: bash -el {0}
-      - name: Run ty-jax
-        run: pixi run --locked ty-jax
+      - name: Run ty (jax backend)
+        run: pixi run --locked -e py314 prek run ty-jax --all-files
         shell: bash -el {0}

.pre-commit-config.yaml

@@ -6,15 +6,15 @@ repos:
       - id: check-useless-excludes
       # - id: identity  # Prints all files passed to pre-commits. Debugging.
   - repo: https://github.com/tox-dev/pyproject-fmt
-    rev: v2.23.0
+    rev: v2.25.0
     hooks:
       - id: pyproject-fmt
   - repo: https://github.com/lyz-code/yamlfix
     rev: 1.19.1
     hooks:
       - id: yamlfix
   - repo: https://github.com/python-jsonschema/check-jsonschema
-    rev: 0.37.2
+    rev: 0.37.3
     hooks:
       - id: check-github-workflows
       - id: check-jsonschema
@@ -59,7 +59,7 @@ repos:
     hooks:
       - id: yamllint
   - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: v0.15.15
+    rev: v0.15.19
     hooks:
       - id: ruff-check
         args:
@@ -73,6 +73,33 @@ repos:
           - jupyter
           - pyi
           - python
+  - repo: https://github.com/astral-sh/ty-pre-commit
+    rev: v0.0.52
+    hooks:
+      - id: ty
+        name: ty (numpy backend)
+        # `--no-project` keeps uv from creating a `.venv`/`uv.lock` in this
+        # pixi-managed repo; ty resolves third-party imports from the pixi
+        # environment configured via `tool.ty.environment.python` in
+        # pyproject.toml (requires `pixi install -e py314` once).
+        args:
+          - --no-project
+  - repo: local
+    hooks:
+      - id: ty-jax
+        name: ty (jax backend)
+        # Same `tool.ty` configuration from pyproject.toml; only the
+        # environment ty resolves third-party imports from is overridden
+        # (requires `pixi install -e py314-jax` once). A local hook because
+        # the official hook's `uv check` does not forward `--python` to ty.
+        # Keep the ty pin in sync with the `ty-pre-commit` rev above.
+        language: python
+        additional_dependencies:
+          - ty==0.0.49
+        entry: ty check --python .pixi/envs/py314-jax
+        pass_filenames: false
+        always_run: true
+        require_serial: true
   - repo: https://github.com/kynan/nbstripout
     rev: 0.9.1
     hooks:
@@ -94,3 +121,8 @@ repos:
         files: (AGENTS\.md|CLAUDE\.md|README\.md|CHANGES\.md|CODE_OF_CONDUCT\.md|modules/.*\.md|profiles/.*\.md)
 ci:
   autoupdate_schedule: monthly
+  # pre-commit.ci blocks network access at hook runtime and has no pixi
+  # environments; the `run-ty` job in GitHub Actions covers type checking.
+  skip:
+    - ty
+    - ty-jax

AGENTS.md

@@ -26,18 +26,19 @@ pixi run -e py314 pytest tests/test_main.py
 # Run a single test
 pixi run -e py314 pytest tests/test_main.py::test_function_name -v
 
-# Type checking with ty
-pixi run -e ty ty
+# Type checking with ty (runs as pre-commit hooks against the py314 and
+# py314-jax environments; requires both to be installed)
+prek run ty --all-files
+prek run ty-jax --all-files
 
-# Available environments: py311, py312, py313, py314, py314-jax, py314-cuda, py314-metal, ty
+# Available environments: py311, py312, py313, py314, py314-jax, py314-cuda, py314-metal
 ```
 
-Before finishing any task that modifies code, always run these three verification steps
-in order:
+Before finishing any task that modifies code, always run these two verification steps in
+order:
 
-1. `pixi run ty` (type checker)
 1. `pixi run -e py314-jax prek run --all-files` (quality checks: linting, formatting,
-   yaml, etc.)
+   yaml, type checking with ty under both backends, etc.)
 1. `pixi run -e py314-jax tests -n 7` (full test suite)
 
 Pre-commit hooks run automatically via pre-commit.ci. Linting uses Ruff with `ALL` rules

CHANGES.md

@@ -10,6 +10,11 @@ releases are available on [Anaconda.org](https://anaconda.org/conda-forge/ttsim)
   (computable from `processed_data` alone) and `raw_results.columns_with_original_p_ids`
   (the reverse-translation of endogenous `p_id_*` columns from {gh}`108`).
   ({ghuser}`MImmesberger`)
+- {gh}`121` Switch the GEP-10 unit checks on: two always-on `fail_if` nodes verify every
+  assembled policy environment — mandatory `unit=` declarations on all active nodes and
+  a conservative dry-run of every function body against its producers' resolved units.
+  Annotate mettsim end to end as the worked example, and check it over all policy dates
+  in CI. ({ghuser}`MImmesberger`)
 - Fill in the `count_by_p_id`, `mean_by_p_id`, `max_by_p_id`, `min_by_p_id`,
   `any_by_p_id`, and `all_by_p_id` aggregations on both the NumPy and JAX backends.
   Negative source `p_id` entries are masked out so they cannot influence the result;