flopscope 0.4.0__tar.gz → 0.4.2__tar.gz

flopscope-0.4.2/.github/workflows/ci.yml

@@ -0,0 +1,435 @@
+name: CI
+
+on:
+  workflow_dispatch:
+    inputs:
+      run_numpy_compat:
+        description: Force NumPy compatibility checks on this run
+        required: false
+        default: false
+        type: boolean
+  push:
+    branches: [main]
+    paths:
+      - 'src/**'
+      - 'tests/**'
+      - 'flopscope-server/**'
+      - 'flopscope-client/**'
+      - 'website/**'
+      - 'scripts/**'
+      - '.github/workflows/ci.yml'
+      - 'pyproject.toml'
+      - 'uv.lock'
+  pull_request:
+    branches: [main]
+    paths:
+      - 'src/**'
+      - 'tests/**'
+      - 'flopscope-server/**'
+      - 'flopscope-client/**'
+      - 'website/**'
+      - 'scripts/**'
+      - '.github/workflows/ci.yml'
+      - 'pyproject.toml'
+      - 'uv.lock'
+
+permissions:
+  contents: read
+  pull-requests: write
+
+concurrency:
+  group: ${{ github.workflow }}-${{ github.head_ref || github.ref_name }}
+  cancel-in-progress: true
+
+env:
+  FLOPSCOPE_DOCS_ROOT: https://aicrowd.github.io/flopscope/docs
+
+jobs:
+  detect-impact:
+    name: Detect changed paths for expensive checks
+    runs-on: ubuntu-latest
+    outputs:
+      run_numpy_compat: ${{ steps.detect.outputs.run_numpy_compat }}
+    steps:
+      - if: github.event_name != 'workflow_dispatch'
+        uses: actions/checkout@v4
+        with:
+          fetch-depth: 0
+      - name: Determine whether to run NumPy compatibility checks
+        id: detect
+        env:
+          GH_TOKEN: ${{ github.token }}
+        run: |
+          set -euo pipefail
+
+          changed_files="$(mktemp)"
+          run_numpy_compat="false"
+
+          if [[ "${{ github.event_name }}" == "pull_request" ]]; then
+            gh api "repos/${{ github.repository }}/pulls/${{ github.event.number }}/files?per_page=200" --paginate --jq '.[].filename' > "$changed_files"
+          elif [[ "${{ github.event_name }}" == "push" ]]; then
+            base="${{ github.event.before }}"
+            head="${{ github.sha }}"
+            if git cat-file -e "${base}^{commit}" 2>/dev/null; then
+              git diff --name-only "$base" "$head" > "$changed_files"
+            elif git rev-parse --verify HEAD~1 >/dev/null 2>&1; then
+              git diff --name-only HEAD~1 "$head" > "$changed_files"
+            else
+              : > "$changed_files"
+            fi
+          elif [[ "${{ github.event_name }}" == "workflow_dispatch" && "${{ inputs.run_numpy_compat }}" == "true" ]]; then
+            run_numpy_compat="true"
+            echo "run_numpy_compat=$run_numpy_compat" >> "$GITHUB_OUTPUT"
+            exit 0
+          fi
+
+          if [[ "$run_numpy_compat" == "false" && -s "$changed_files" ]]; then
+            while IFS= read -r file; do
+              case "$file" in
+                src/*|tests/*|flopscope-server/*|flopscope-client/*|scripts/*|website/*|pyproject.toml|uv.lock|.github/workflows/ci.yml)
+                  run_numpy_compat="true"
+                  break
+                  ;;
+              esac
+            done < "$changed_files"
+          fi
+
+          echo "run_numpy_compat=$run_numpy_compat" >> "$GITHUB_OUTPUT"
+
+  lint:
+    runs-on: ubuntu-latest
+    timeout-minutes: 20
+    steps:
+      - uses: actions/checkout@v4
+        with:
+          fetch-depth: 0
+
+      - name: Install ruff and gitlint
+        run: pip install ruff gitlint
+
+      - name: Ruff lint check
+        run: ruff check .
+
+      - name: Ruff format check
+        run: ruff format --check .
+
+      - name: Conventional commits (gitlint)
+        run: gitlint --commits origin/main..HEAD
+
+  typecheck:
+    name: Pyright
+    runs-on: ubuntu-latest
+    timeout-minutes: 30
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Install uv
+        uses: astral-sh/setup-uv@v4
+
+      - name: Set up Python
+        run: uv python install 3.10
+
+      - name: Install dependencies
+        run: uv sync --all-extras
+
+      - name: Pyright
+        run: uv run pyright src/flopscope tests
+
+  test:
+    needs: [detect-impact]
+    runs-on: ubuntu-latest
+    timeout-minutes: 120
+    strategy:
+      fail-fast: true
+      matrix:
+        python-version: ["3.10", "3.11", "3.12"]
+        numpy-version: ["2.0", "2.1", "2.2"]
+        include:
+          - python-version: "3.12"
+            numpy-version: "2.3"
+          - python-version: "3.12"
+            numpy-version: "2.4"
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Install uv
+        uses: astral-sh/setup-uv@v4
+
+      - name: Set up Python ${{ matrix.python-version }}
+        run: uv python install ${{ matrix.python-version }}
+
+      - name: Set up Node.js (for JS cross-validation)
+        uses: actions/setup-node@v4
+        with:
+          node-version: '20'
+
+      - name: Install dependencies
+        run: uv sync --all-extras
+
+      - name: Pin NumPy ${{ matrix.numpy-version }}
+        run: uv pip install 'numpy~=${{ matrix.numpy-version }}.0'
+
+      - name: Run tests with coverage
+        run: uv run pytest --cov=flopscope --cov-fail-under=85
+
+  numpy-compat:
+    if: needs.detect-impact.outputs.run_numpy_compat == 'true'
+    needs: [detect-impact]
+    runs-on: ubuntu-latest
+    timeout-minutes: 150
+    strategy:
+      fail-fast: true
+      matrix:
+        python-version: ["3.12"]
+        numpy-version: ["2.2", "2.3", "2.4"]
+    steps:
+      - uses: actions/checkout@v4
+        with:
+          fetch-depth: 0
+
+      - name: Install uv
+        uses: astral-sh/setup-uv@v4
+
+      - name: Set up Python ${{ matrix.python-version }}
+        run: uv python install ${{ matrix.python-version }}
+
+      - name: Set up Node.js (for JS cross-validation)
+        uses: actions/setup-node@v4
+        with:
+          node-version: '20'
+
+      - name: Install dependencies
+        run: uv sync --all-extras
+
+      - name: Pin NumPy ${{ matrix.numpy-version }}
+        run: uv pip install 'numpy~=${{ matrix.numpy-version }}.0'
+
+      - name: Run tests with coverage
+        run: uv run pytest --cov=flopscope --cov-fail-under=85
+
+      - name: "NumPy compat: core umath"
+        run: uv run pytest tests/numpy_compat/ --pyargs numpy._core.tests.test_umath -n auto -q
+
+      - name: "NumPy compat: core ufunc"
+        run: uv run pytest tests/numpy_compat/ --pyargs numpy._core.tests.test_ufunc -n auto -q
+
+      - name: "NumPy compat: core numeric"
+        run: uv run pytest tests/numpy_compat/ --pyargs numpy._core.tests.test_numeric -n auto -q -k "not test_for_reference_leak"
+
+      - name: "NumPy compat: linalg"
+        run: uv run pytest tests/numpy_compat/ --pyargs numpy.linalg.tests.test_linalg -n auto -q
+
+      - name: "NumPy compat: fft pocketfft"
+        run: uv run pytest tests/numpy_compat/ --pyargs numpy.fft.tests.test_pocketfft -n auto -q
+
+      - name: "NumPy compat: fft helper"
+        run: uv run pytest tests/numpy_compat/ --pyargs numpy.fft.tests.test_helper -n auto -q
+
+      - name: "NumPy compat: polynomial"
+        run: uv run pytest tests/numpy_compat/ --pyargs numpy.polynomial.tests.test_polynomial -n auto -q
+
+      - name: "NumPy compat: random"
+        run: uv run pytest tests/numpy_compat/ --pyargs numpy.random.tests.test_random -n auto -q
+
+  client-server-sync:
+    runs-on: ubuntu-latest
+    timeout-minutes: 45
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Install uv
+        uses: astral-sh/setup-uv@v4
+
+      - name: Set up Python
+        run: uv python install 3.12
+
+      - name: Install dependencies
+        run: uv sync --all-extras
+
+      - name: Check generated files are in sync
+        run: uv run python scripts/sync_client.py --check
+
+      - name: Check package versions are in lockstep
+        run: uv run python scripts/check_version_sync.py
+
+      - name: Run parity tests
+        run: uv run pytest tests/test_client_server_parity.py tests/test_serialization_parity.py -v
+
+  client-server-integration:
+    needs: [client-server-sync]
+    runs-on: ubuntu-latest
+    timeout-minutes: 90
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Install uv
+        uses: astral-sh/setup-uv@v4
+
+      - name: Set up Python
+        run: uv python install 3.12
+
+      - name: Install core + server dependencies
+        run: |
+          uv sync --all-extras
+          cd flopscope-server && uv sync
+
+      - name: Run integration tests
+        run: |
+          cd flopscope-client && uv run pytest tests/test_full_integration.py -v --tb=short
+
+  website-tests:
+    runs-on: ubuntu-latest
+    timeout-minutes: 120
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Install uv
+        uses: astral-sh/setup-uv@v4
+
+      - name: Set up Python
+        run: uv python install 3.12
+
+      - name: Install Python dependencies
+        run: uv sync --all-extras
+
+      - name: Set up Node.js
+        uses: actions/setup-node@v4
+        with:
+          node-version: '20'
+          cache: 'npm'
+          cache-dependency-path: website/package-lock.json
+
+      - name: Install website dependencies
+        run: cd website && npm ci
+
+      - name: Generate API data
+        run: uv run python scripts/generate_api_docs.py
+
+      - name: Run website regression tests
+        run: cd website && npm test
+
+      - name: Build website
+        run: cd website && npm run build
+
+      - name: Run GH Pages compatibility checks
+        run: cd website && npm run check:gh-pages
+
+  docs:
+    if: github.event_name == 'push' && github.ref == 'refs/heads/main'
+    needs: [lint, test, website-tests]
+    runs-on: ubuntu-latest
+    timeout-minutes: 90
+    permissions:
+      contents: write
+    steps:
+      - uses: actions/checkout@v4
+        with:
+          fetch-depth: 0
+
+      - name: Install uv
+        uses: astral-sh/setup-uv@v4
+
+      - name: Set up Python
+        run: uv python install 3.12
+
+      - name: Install Python dependencies
+        run: uv sync --all-extras
+
+      - name: Set up Node.js
+        uses: actions/setup-node@v4
+        with:
+          node-version: '20'
+          cache: 'npm'
+          cache-dependency-path: website/package-lock.json
+
+      - name: Install website dependencies
+        run: cd website && npm ci
+
+      - name: Generate API data
+        run: uv run python scripts/generate_api_docs.py
+
+      - name: Verify generated API data
+        run: uv run python scripts/generate_api_docs.py --verify
+
+      - name: Build website
+        run: cd website && npm run build
+
+      - name: Run GH Pages compatibility checks
+        run: cd website && npm run check:gh-pages
+
+  ci-cost-report:
+    name: CI cost estimate
+    if: always()
+    needs:
+      - lint
+      - typecheck
+      - test
+      - numpy-compat
+      - client-server-sync
+      - client-server-integration
+      - website-tests
+      - docs
+    runs-on: ubuntu-latest
+    timeout-minutes: 10
+    steps:
+      - name: Summarize elapsed job times and estimated costs
+        env:
+          GH_TOKEN: ${{ github.token }}
+          RATE_PER_MINUTE: "0.006"
+        run: |
+          set -euo pipefail
+
+          jobs_json="$(mktemp)"
+          if ! gh api "repos/${GITHUB_REPOSITORY}/actions/runs/${GITHUB_RUN_ID}/jobs?per_page=200" --paginate > "$jobs_json"; then
+            echo "Unable to fetch jobs API for this run; skipping cost report."
+            exit 0
+          fi
+
+          summary_file="$(mktemp)"
+          python - "$jobs_json" <<'PY' > "$summary_file"
+          import datetime
+          import json
+          import os
+          import sys
+
+          rate = float(os.getenv("RATE_PER_MINUTE", "0.006"))
+
+          with open(sys.argv[1], "r", encoding="utf-8") as f:
+              data = json.load(f)
+
+          jobs = data.get("jobs", [])
+          rows = []
+          total_minutes = 0.0
+
+          for job in jobs:
+              started_at = job.get("started_at")
+              completed_at = job.get("completed_at")
+              name = job.get("name") or f"job-{job.get('id')}"
+
+              if not started_at or not completed_at:
+                  continue
+
+              started_dt = datetime.datetime.fromisoformat(started_at.replace("Z", "+00:00"))
+              completed_dt = datetime.datetime.fromisoformat(completed_at.replace("Z", "+00:00"))
+              elapsed_minutes = max((completed_dt - started_dt).total_seconds() / 60.0, 0.0)
+              rows.append((name, elapsed_minutes))
+              total_minutes += elapsed_minutes
+
+          rows.sort(key=lambda row: row[1], reverse=True)
+          total_cost = total_minutes * rate
+
+          print("## CI cost estimate")
+          print()
+          print(f"Estimated workflow total: `{total_minutes:.1f}` minutes")
+          print(f"Estimated billed cost at `${rate:.3f}`/minute: `${total_cost:.2f}`")
+          print()
+          print("### Job timings")
+          for name, minutes in rows:
+              print(f"- `{name}`: `{minutes:.2f}` min (`${minutes * rate:.4f}`)")
+          PY
+
+          echo "### CI cost report ###"
+          cat "$summary_file"
+          if [[ "${{ github.event_name }}" == "pull_request" ]]; then
+            gh issue comment "${{ github.event.number }}" --body-file "$summary_file" || true
+          fi

{flopscope-0.4.0 → flopscope-0.4.2}/CHANGELOG.md

@@ -1,5 +1,45 @@
 # Changelog
 
+## v0.4.2 (2026-06-01)
+
+### Feat
+
+- **ci**: gate numpy compat checks
+
+### Fix
+
+- support fnp.random.default_rng() across the client/server boundary
+
+## v0.4.1 (2026-05-26)
+
+Bug-fix release for the broken `flopscope[server]` extra in v0.4.0.
+
+### Fixed
+
+- The `flopscope[server]` extra now correctly pins
+  `flopscope-server==0.4.1` (matching the rest of the release). In
+  v0.4.0 the extra was stuck at `flopscope-server==0.3.0` because the
+  pin location was not tracked by commitizen's `version_files`, so
+  `pip install "flopscope[server]==0.4.0"` from PyPI was
+  **unresolvable** (it pulled flopscope-server 0.3.0, which in turn
+  requires flopscope==0.3.0, conflicting with the 0.4.0 root).
+- `pip install "flopscope[server]==0.4.1"` resolves cleanly.
+
+### Tooling
+
+- `commitizen.version_files` now includes
+  `pyproject.toml:flopscope-server==` so the `[server]` extra pin
+  follows future bumps automatically.
+- `scripts/check_version_sync.py` now compares 8 version locations
+  (added the `[server]` extra pin) and would catch this regression
+  in CI. `tests/test_check_version_sync.py` includes a corresponding
+  guard test (`test_server_extra_pin_drift_detected`).
+- Drift-detection tests in `tests/test_check_version_sync.py` are
+  now version-agnostic (they read the current X.Y.Z from
+  `pyproject.toml` at test time instead of hardcoding it). v0.4.0's
+  main CI failed after the bump because hardcoded `"0.3.0"` strings
+  no longer matched.
+
 ## v0.4.0 (2026-05-26)
 
 Follow-up to v0.3.0 that completes the multi-package PyPI release. All

{flopscope-0.4.0 → flopscope-0.4.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flopscope
-Version: 0.4.0
+Version: 0.4.2
 Summary: NumPy-compatible math primitives with FLOP counting for the Mechanistic Estimation Challenge
 License-Expression: MIT
 Requires-Python: >=3.10
@@ -20,7 +20,7 @@ Requires-Dist: scipy>=1.10; extra == 'dev'
 Provides-Extra: docs
 Requires-Dist: numpydoc>=1.10; extra == 'docs'
 Provides-Extra: server
-Requires-Dist: flopscope-server==0.3.0; extra == 'server'
+Requires-Dist: flopscope-server==0.4.2; extra == 'server'
 Provides-Extra: sympy
 Requires-Dist: sympy>=1.11; extra == 'sympy'
 Description-Content-Type: text/markdown

{flopscope-0.4.0 → flopscope-0.4.2}/flopscope-client/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "flopscope-client"
-version = "0.4.0"
+version = "0.4.2"
 description = "Lightweight flopscope client — drop-in replacement with no numpy dependency"
 readme = "README.md"
 requires-python = ">=3.10"

{flopscope-0.4.0 → flopscope-0.4.2}/flopscope-client/src/flopscope/__init__.py

@@ -18,7 +18,7 @@ import builtins
 import struct
 from typing import Any
 
-__version__ = "0.4.0"
+__version__ = "0.4.2"
 
 # ---------------------------------------------------------------------------
 # Errors

{flopscope-0.4.0 → flopscope-0.4.2}/flopscope-client/src/flopscope/_remote_array.py

@@ -597,6 +597,85 @@ class RemoteArray(metaclass=_RemoteArrayMeta):
     __hash__ = None  # type: ignore[assignment]
 
 
+# ---------------------------------------------------------------------------
+# RemoteGenerator
+# ---------------------------------------------------------------------------
+
+
+class RemoteGenerator:
+    """Transparent proxy for a server-side numpy ``Generator``.
+
+    ``fnp.random.default_rng(seed)`` returns one of these. Sampling methods
+    dispatch to the server, where the RNG state lives and advances — so the
+    stream is deterministic per seed and FLOP-counted server-side, exactly
+    like the in-process counted Generator.
+    """
+
+    __slots__ = ("_handle_id",)
+
+    def __init__(self, handle_id: str) -> None:
+        self._handle_id = handle_id
+
+    @property
+    def handle_id(self) -> str:
+        return self._handle_id
+
+    def __repr__(self) -> str:
+        return f"RemoteGenerator(handle_id={self._handle_id!r})"
+
+    def _call(self, method: str, *args: Any, **kwargs: Any) -> Any:
+        from flopscope._connection import get_connection
+        from flopscope._protocol import encode_request
+
+        encoded_args = [_encode_arg(self)] + [_encode_arg(a) for a in args]
+        encoded_kwargs = {k: _encode_arg(v) for k, v in kwargs.items()}
+        resp = get_connection().send_recv(
+            encode_request(
+                f"Generator.{method}", args=encoded_args, kwargs=encoded_kwargs
+            )
+        )
+        return _result_from_response(resp)
+
+    def uniform(self, *args, **kwargs):
+        return self._call("uniform", *args, **kwargs)
+
+    def standard_normal(self, *args, **kwargs):
+        return self._call("standard_normal", *args, **kwargs)
+
+    def normal(self, *args, **kwargs):
+        return self._call("normal", *args, **kwargs)
+
+    def integers(self, *args, **kwargs):
+        return self._call("integers", *args, **kwargs)
+
+    def random(self, *args, **kwargs):
+        return self._call("random", *args, **kwargs)
+
+    def standard_exponential(self, *args, **kwargs):
+        return self._call("standard_exponential", *args, **kwargs)
+
+    def exponential(self, *args, **kwargs):
+        return self._call("exponential", *args, **kwargs)
+
+    def poisson(self, *args, **kwargs):
+        return self._call("poisson", *args, **kwargs)
+
+    def binomial(self, *args, **kwargs):
+        return self._call("binomial", *args, **kwargs)
+
+    def beta(self, *args, **kwargs):
+        return self._call("beta", *args, **kwargs)
+
+    def gamma(self, *args, **kwargs):
+        return self._call("gamma", *args, **kwargs)
+
+    def choice(self, *args, **kwargs):
+        return self._call("choice", *args, **kwargs)
+
+    def permutation(self, *args, **kwargs):
+        return self._call("permutation", *args, **kwargs)
+
+
 # ---------------------------------------------------------------------------
 # Module-level helper
 # ---------------------------------------------------------------------------
@@ -614,6 +693,9 @@ def _result_from_response(resp: dict) -> RemoteArray | RemoteScalar | tuple | di
     """
     result = resp.get("result", {})
 
+    if "gen_id" in result:
+        return RemoteGenerator(result["gen_id"])
+
     if "value" in result:
         return RemoteScalar(value=result["value"], dtype=result.get("dtype", "float64"))
 
@@ -682,6 +764,8 @@ def _encode_arg(arg):
         return arg._value
     if isinstance(arg, RemoteArray):
         return {"__handle__": arg.handle_id}
+    if isinstance(arg, RemoteGenerator):
+        return {"__gen__": arg.handle_id}
     from flopscope._perm_group import SymmetryGroup
 
     if isinstance(arg, SymmetryGroup):