ergmx 0.1.0__tar.gz

ergmx-0.1.0/.github/actions/test-wheel/action.yml

@@ -0,0 +1,13 @@
+name: Test the wheel
+description: Install the built wheel in fresh environments, without Rust or the source tree, and fit a model.
+runs:
+  using: composite
+  steps:
+  - uses: astral-sh/setup-uv@c18668ad3cf93ea998bef934396af7bb5c839dc7  # v10.2.0
+  - name: Fit a model with the wheel, on the oldest and newest Python
+    shell: bash
+    run: |
+      for python in 3.11 3.14; do
+        uv run --isolated --no-project --python "$python" --with "$(ls dist/*.whl)" --with igraph \
+          python -c "import ergmx; from ergmx import datasets; fit = ergmx.ergm(datasets.load('samplk3'), 'edges + mutual', seed=1); assert fit.converged; print('Python $python:', fit.coef)"
+      done

ergmx-0.1.0/.github/workflows/docs.yml

@@ -0,0 +1,56 @@
+# This workflow builds the documentation with Sphinx and publishes it to GitHub Pages
+# For more information see: https://docs.github.com/en/pages/getting-started-with-github-pages/using-custom-workflows-with-github-pages
+
+name: Documentation
+
+on:
+  push:
+    branches: [main]
+  pull_request:
+  workflow_dispatch:
+
+permissions:
+  contents: read
+
+concurrency:
+  group: pages-${{ github.ref }}
+  cancel-in-progress: false
+
+jobs:
+  build:
+
+    runs-on: ubuntu-latest
+
+    steps:
+    - uses: actions/checkout@v7
+    - uses: astral-sh/setup-uv@c18668ad3cf93ea998bef934396af7bb5c839dc7  # v10.2.0
+      with:
+        enable-cache: true
+    - name: Install the Rust toolchain of rust-toolchain.toml
+      run: rustup toolchain install
+    - name: Build ergmx and install the documentation tools
+      run: uv sync --locked --group docs
+    - name: Build documentation
+      run: uv run sphinx-build -W --keep-going -d docs/_build/doctrees docs docs/_build/html
+    - name: Upload documentation
+      uses: actions/upload-pages-artifact@v5
+      with:
+        path: docs/_build/html
+
+  # Pull requests only check that the documentation builds.
+  deploy:
+
+    if: github.event_name != 'pull_request'
+    needs: build
+    runs-on: ubuntu-latest
+    permissions:
+      pages: write
+      id-token: write
+    environment:
+      name: github-pages
+      url: ${{ steps.deployment.outputs.page_url }}
+
+    steps:
+    - name: Deploy to GitHub Pages
+      id: deployment
+      uses: actions/deploy-pages@v5

ergmx-0.1.0/.github/workflows/release.yml

@@ -0,0 +1,124 @@
+# Builds ergmx's wheels, with the Rust core compiled in, and its source
+# distribution, and publishes them to PyPI when a GitHub release is published.
+#
+# The wheels use Python's stable ABI (abi3): one wheel per platform serves
+# every Python from 3.11 on. Run it by hand (workflow_dispatch) to build and
+# test the wheels without publishing them.
+#
+# Publishing uses PyPI's trusted publishing, which needs a one-time setup on
+# PyPI: see docs/user-guide/installation.md, "Publishing a release".
+
+name: Release
+
+on:
+  release:
+    types: [published]
+  workflow_dispatch:
+
+permissions:
+  contents: read
+
+jobs:
+  linux:
+    name: Linux ${{ matrix.libc }} ${{ matrix.target }}
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        target: [x86_64, aarch64]
+        libc: [manylinux, musllinux]
+    steps:
+    - uses: actions/checkout@v7
+    - name: Build the wheel
+      uses: PyO3/maturin-action@v1
+      with:
+        target: ${{ matrix.target }}
+        manylinux: ${{ matrix.libc == 'manylinux' && 'auto' || 'musllinux_1_2' }}
+        args: --release --out dist
+    - name: Test the wheel
+      if: matrix.target == 'x86_64' && matrix.libc == 'manylinux'
+      uses: ./.github/actions/test-wheel
+    - uses: actions/upload-artifact@v7
+      with:
+        name: wheels-linux-${{ matrix.libc }}-${{ matrix.target }}
+        path: dist
+
+  macos:
+    name: macOS ${{ matrix.target }}
+    runs-on: ${{ matrix.runner }}
+    strategy:
+      matrix:
+        include:
+        - runner: macos-15-intel
+          target: x86_64
+        - runner: macos-latest
+          target: aarch64
+    steps:
+    - uses: actions/checkout@v7
+    - name: Build the wheel
+      uses: PyO3/maturin-action@v1
+      with:
+        target: ${{ matrix.target }}
+        args: --release --out dist
+    - name: Test the wheel
+      uses: ./.github/actions/test-wheel
+    - uses: actions/upload-artifact@v7
+      with:
+        name: wheels-macos-${{ matrix.target }}
+        path: dist
+
+  windows:
+    name: Windows x64
+    runs-on: windows-latest
+    steps:
+    - uses: actions/checkout@v7
+    - name: Build the wheel
+      uses: PyO3/maturin-action@v1
+      with:
+        target: x64
+        args: --release --out dist
+    - name: Test the wheel
+      uses: ./.github/actions/test-wheel
+    - uses: actions/upload-artifact@v7
+      with:
+        name: wheels-windows-x64
+        path: dist
+
+  sdist:
+    name: Source distribution
+    runs-on: ubuntu-latest
+    steps:
+    - uses: actions/checkout@v7
+    - name: Build the sdist
+      uses: PyO3/maturin-action@v1
+      with:
+        command: sdist
+        args: --out dist
+    - uses: actions/upload-artifact@v7
+      with:
+        name: wheels-sdist
+        path: dist
+
+  publish:
+    name: Publish to PyPI
+    if: github.event_name == 'release'
+    needs: [linux, macos, windows, sdist]
+    runs-on: ubuntu-latest
+    environment:
+      name: pypi
+      url: https://pypi.org/project/ergmx/
+    permissions:
+      id-token: write      # trusted publishing
+      attestations: write  # build provenance
+    steps:
+    - uses: actions/download-artifact@v8
+      with:
+        pattern: wheels-*
+        merge-multiple: true
+        path: dist
+    - name: Attest build provenance
+      uses: actions/attest-build-provenance@v4
+      with:
+        subject-path: dist/*
+    - uses: astral-sh/setup-uv@c18668ad3cf93ea998bef934396af7bb5c839dc7  # v10.2.0
+    - name: Publish
+      run: uv publish --trusted-publishing always dist/*

ergmx-0.1.0/.github/workflows/tests.yml

@@ -0,0 +1,65 @@
+# Runs the test suites on every push to main and every pull request.
+
+name: Tests
+
+on:
+  push:
+    branches: [main]
+  pull_request:
+  workflow_dispatch:
+
+permissions:
+  contents: read
+
+jobs:
+  python:
+    name: ${{ matrix.os }}, Python ${{ matrix.python }}
+    runs-on: ${{ matrix.os }}
+    strategy:
+      fail-fast: false
+      matrix:
+        os: [ubuntu-latest, macos-latest, windows-latest]
+        python: ["3.11", "3.14"]
+    steps:
+    - uses: actions/checkout@v7
+    - uses: astral-sh/setup-uv@c18668ad3cf93ea998bef934396af7bb5c839dc7  # v10.2.0
+      with:
+        python-version: ${{ matrix.python }}
+        enable-cache: true
+    - name: Install the Rust toolchain of rust-toolchain.toml
+      run: rustup toolchain install
+    - name: Build ergmx and install the development dependencies
+      run: uv sync --locked
+    - name: Test
+      run: uv run pytest
+
+  lowest:
+    name: Oldest supported dependencies
+    runs-on: ubuntu-latest
+    steps:
+    - uses: actions/checkout@v7
+    - uses: astral-sh/setup-uv@c18668ad3cf93ea998bef934396af7bb5c839dc7  # v10.2.0
+    - name: Install the Rust toolchain of rust-toolchain.toml
+      run: rustup toolchain install
+    - name: Install ergmx with the oldest versions its requirements allow
+      run: |
+        uv venv --python 3.11
+        uv pip install --resolution lowest-direct . "pytest>=8" "igraph>=0.11.5" "networkx>=3.2" "matplotlib>=3.9"
+    - name: Test
+      run: .venv/bin/python -m pytest
+
+  rust:
+    name: Rust
+    runs-on: ubuntu-latest
+    steps:
+    - uses: actions/checkout@v7
+    # The Rust tests link to libpython, which this Python provides.
+    - uses: actions/setup-python@v7
+      with:
+        python-version: "3.14"
+    - name: Install the Rust toolchain of rust-toolchain.toml
+      run: rustup toolchain install
+    - name: Clippy
+      run: cargo clippy --release --all-targets -- -D warnings
+    - name: Test
+      run: cargo test --release

ergmx-0.1.0/.gitignore

@@ -0,0 +1,9 @@
+target/
+.venv/
+__pycache__/
+*.so
+*.pyd
+.pytest_cache/
+dist/
+docs/_build/
+docs/generated/

ergmx-0.1.0/.python-version

@@ -0,0 +1 @@
+3.14

ergmx-0.1.0/Cargo.lock

@@ -0,0 +1,281 @@
+# This file is automatically @generated by Cargo.
+# It is not intended for manual editing.
+version = 4
+
+[[package]]
+name = "autocfg"
+version = "1.5.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f2032f911046de80f0a198e0901378627c33f59ea0ac00e363d481118bd70a53"
+
+[[package]]
+name = "crossbeam-deque"
+version = "0.8.8"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "622f3fc73690be383c7214310406f28a90e6edeadc3cea882f9d71e495b9711a"
+dependencies = [
+ "crossbeam-epoch",
+ "crossbeam-utils",
+]
+
+[[package]]
+name = "crossbeam-epoch"
+version = "0.9.21"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "dc74980687109a3b14c72fd458107bf0baa1da1a1a805e178d15501ba9b86d9d"
+dependencies = [
+ "crossbeam-utils",
+]
+
+[[package]]
+name = "crossbeam-utils"
+version = "0.8.23"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "a31eee39dddec8330830986fcd7625edb5a24ec90ea038215273bbc3adb08ac6"
+
+[[package]]
+name = "either"
+version = "1.18.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "252afb9ae5eaa683babdc6a068b3f5726eb19e05070c731f9b2a23a7c3e8ed34"
+
+[[package]]
+name = "ergmx"
+version = "0.1.0"
+dependencies = [
+ "numpy",
+ "pyo3",
+ "rayon",
+ "rustc-hash",
+]
+
+[[package]]
+name = "heck"
+version = "0.5.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "2304e00983f87ffb38b55b444b5e3b60a884b5d30c0fca7d82fe33449bbe55ea"
+
+[[package]]
+name = "libc"
+version = "0.2.189"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "3eaf3ede3fee6db1a4c2ee091bf8a8b4dccdc6d17f656fb07896ee72867612f2"
+
+[[package]]
+name = "matrixmultiply"
+version = "0.3.11"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "3f607c237553f086e7043417a51df26b2eb899d3caff94e6a67592ff992fedc7"
+dependencies = [
+ "autocfg",
+ "rawpointer",
+]
+
+[[package]]
+name = "ndarray"
+version = "0.17.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "520080814a7a6b4a6e9070823bb24b4531daac8c4627e08ba5de8c5ef2f2752d"
+dependencies = [
+ "matrixmultiply",
+ "num-complex",
+ "num-integer",
+ "num-traits",
+ "portable-atomic",
+ "portable-atomic-util",
+ "rawpointer",
+]
+
+[[package]]
+name = "num-complex"
+version = "0.4.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "73f88a1307638156682bada9d7604135552957b7818057dcef22705b4d509495"
+dependencies = [
+ "num-traits",
+]
+
+[[package]]
+name = "num-integer"
+version = "0.1.47"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "7ce2d95d4b3734dc35aa2f45e1aa22cd416814592a4f9d9205e11affd5b8e10b"
+dependencies = [
+ "num-traits",
+]
+
+[[package]]
+name = "num-traits"
+version = "0.2.19"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "071dfc062690e90b734c0b2273ce72ad0ffa95f0c74596bc250dcfd960262841"
+dependencies = [
+ "autocfg",
+]
+
+[[package]]
+name = "numpy"
+version = "0.29.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "6a5b15d63a5ff39e378daed0e1340d3a5964703ea9712eb09a0dc66fade996f4"
+dependencies = [
+ "libc",
+ "ndarray",
+ "num-complex",
+ "num-integer",
+ "num-traits",
+ "pyo3",
+ "pyo3-build-config",
+ "rustc-hash",
+]
+
+[[package]]
+name = "once_cell"
+version = "1.21.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "9f7c3e4beb33f85d45ae3e3a1792185706c8e16d043238c593331cc7cd313b50"
+
+[[package]]
+name = "portable-atomic"
+version = "1.15.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "05c8b63e8d9609db387f0324918f81d68fe27748f084ef092fb35954d0539a85"
+
+[[package]]
+name = "portable-atomic-util"
+version = "0.2.8"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "10ab3eb7f3becc3a1cbc4f2c6f20267996cfc1a6467a873763411b136a122715"
+dependencies = [
+ "portable-atomic",
+]
+
+[[package]]
+name = "proc-macro2"
+version = "1.0.107"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "985e7ec9bb745e6ce6535b544d84d6cd6f7ad8bd711c398938ae983b91a766d9"
+dependencies = [
+ "unicode-ident",
+]
+
+[[package]]
+name = "pyo3"
+version = "0.29.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "4688ddedf473e32662b9b067670129a8afb8c18e351482c70d62ba4a88171e8b"
+dependencies = [
+ "libc",
+ "once_cell",
+ "portable-atomic",
+ "pyo3-build-config",
+ "pyo3-ffi",
+ "pyo3-macros",
+]
+
+[[package]]
+name = "pyo3-build-config"
+version = "0.29.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f41027e41b4bd03f6e60f9f417fe24a6341a6bb744edd62b6f709f2a52ea30e9"
+dependencies = [
+ "target-lexicon",
+]
+
+[[package]]
+name = "pyo3-ffi"
+version = "0.29.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e591a95526fead067432c3b3a33fc74770b87b1e04e73671090d9c2055a2b327"
+dependencies = [
+ "libc",
+ "pyo3-build-config",
+]
+
+[[package]]
+name = "pyo3-macros"
+version = "0.29.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "73225868fc1cd84eef2c3c230ddb91273bf1de46aeb8a4248da76d32a0924a1c"
+dependencies = [
+ "proc-macro2",
+ "pyo3-macros-backend",
+ "quote",
+ "syn",
+]
+
+[[package]]
+name = "pyo3-macros-backend"
+version = "0.29.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "571575aa3749fa6216757dd47d2a3e7ef360f329a40f0666a9fbd14889024952"
+dependencies = [
+ "heck",
+ "proc-macro2",
+ "quote",
+ "syn",
+]
+
+[[package]]
+name = "quote"
+version = "1.0.47"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1fbf4db142a473a8d80c26bbf18454ed458bf8d26c8219c331daecfdbd079001"
+dependencies = [
+ "proc-macro2",
+]
+
+[[package]]
+name = "rawpointer"
+version = "0.2.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "60a357793950651c4ed0f3f52338f53b2f809f32d83a07f72909fa13e4c6c1e3"
+
+[[package]]
+name = "rayon"
+version = "1.12.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "fb39b166781f92d482534ef4b4b1b2568f42613b53e5b6c160e24cfbfa30926d"
+dependencies = [
+ "either",
+ "rayon-core",
+]
+
+[[package]]
+name = "rayon-core"
+version = "1.13.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "22e18b0f0062d30d4230b2e85ff77fdfe4326feb054b9783a3460d8435c8ab91"
+dependencies = [
+ "crossbeam-deque",
+ "crossbeam-utils",
+]
+
+[[package]]
+name = "rustc-hash"
+version = "2.1.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "6b1e7f9a428571be2dc5bc0505c13fb6bf936822b894ec87abf8a08a4e51742d"
+
+[[package]]
+name = "syn"
+version = "2.0.119"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "872831b642d1a07999a962a351ed35b955ea2cfc8f3862091e2a240a84f17297"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "unicode-ident",
+]
+
+[[package]]
+name = "target-lexicon"
+version = "0.13.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "adb6935a6f5c20170eeceb1a3835a49e12e19d792f6dd344ccc76a985ca5a6ca"
+
+[[package]]
+name = "unicode-ident"
+version = "1.0.26"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d245f478577f809a851594d02313b640fb437e0bb33866753cff937863096954"

ergmx-0.1.0/Cargo.toml

@@ -0,0 +1,23 @@
+[package]
+name = "ergmx"
+version = "0.1.0"
+edition = "2024"
+# The oldest Rust that builds ergmx; rust-toolchain.toml pins the one used.
+rust-version = "1.88"
+license = "GPL-3.0-only"
+publish = false
+readme = "README.md"
+
+[lib]
+name = "_core"
+crate-type = ["cdylib"]
+
+[dependencies]
+numpy = "0.29"
+pyo3 = { version = "0.29", features = ["abi3-py311"] }
+rayon = "1.12"
+rustc-hash = "2.1"
+
+[profile.release]
+codegen-units = 1
+lto = "fat"