anemoi-utils 0.4.5__tar.gz → 0.4.40__tar.gz

anemoi_utils-0.4.40/.github/CODEOWNERS

@@ -0,0 +1,7 @@
+# Workflows
+/.github/ @ecmwf/AnemoiSecurity
+
+# Project configs
+/pyproject.toml @ecmwf/AnemoiSecurity
+/.pre-commit-config.yaml @ecmwf/AnemoiSecurity
+/.release-please-config.json @ecmwf/AnemoiSecurity

anemoi_utils-0.4.40/.github/dependabot.yml

@@ -0,0 +1,10 @@
+version: 2
+updates:
+- package-ecosystem: "pip"
+  directory: "/"
+  schedule:
+    interval: "weekly"
+# - package-ecosystem: "github-actions"
+#   directory: "/"
+#   schedule:
+#     interval: "monthly"

anemoi_utils-0.4.40/.github/labeler.yml

@@ -0,0 +1,43 @@
+# This is the configuration file for the labeler action.
+# It assigns labels to pull requests based on the files changed in the PR.
+# See more here: https://github.com/actions/labeler
+dependencies:
+- changed-files:
+  - any-glob-to-any-file:
+    - "**/requirements.txt"
+    - "**/setup.py"
+    - "**/pyproject.toml"
+    - "**/Pipfile"
+    - "**/Pipfile.lock"
+    - "**/requirements/*.txt"
+    - "**/requirements/*.in"
+
+documentation:
+- changed-files:
+  - any-glob-to-any-file:
+    - "**/docs/**/*"
+    - "*.md"
+    - "*.rst"
+
+config:
+- changed-files:
+  - any-glob-to-any-file:
+    - "**/src/**/config/**/*"
+    - "**/src/anemoi/inference/config.py"
+
+CI/CD:
+- changed-files:
+  - any-glob-to-any-file:
+    - "**/.pre-commit-config.yaml"
+    - ".github/**/*"
+    - "tox.ini"
+    - ".coveragerc"
+
+tests:
+- changed-files:
+  - any-glob-to-any-file:
+    - "**/tests/**/*"
+    - "**/test/**/*"
+    - "**/test_*.py"
+    - "**/test.py"
+    - "**/conftest.py"

anemoi_utils-0.4.40/.github/pull_request_template.md

@@ -0,0 +1,15 @@
+## Description
+<!-- What issue or task does this change relate to? -->
+
+## What problem does this change solve?
+<!-- Describe if it's a bugfix, new feature, doc update, or breaking change -->
+
+## What issue or task does this change relate to?
+<!-- link to Issue Number -->
+
+##  Additional notes ##
+<!-- Include any additional information, caveats, or considerations that the reviewer should be aware of. -->
+
+***As a contributor to the Anemoi framework, please ensure that your changes include unit tests, updates to any affected dependencies and documentation, and have been tested in a parallel setting  (i.e., with multiple GPUs). As a reviewer, you are also responsible for verifying these aspects and requesting changes if they are not adequately addressed. For guidelines about those please refer to https://anemoi.readthedocs.io/en/latest/***
+
+By opening this pull request, I affirm that all authors agree to the [Contributor License Agreement.](https://github.com/ecmwf/codex/blob/main/Legal/contributor_license_agreement.md)

anemoi_utils-0.4.40/.github/workflows/downstream-ci-hpc.yml

@@ -0,0 +1,60 @@
+# This workflow triggers tests on dependent packages.
+# The dependency tree itself is defined in ecmwf/downstream-ci/
+name: Test downstream dependent packages
+
+on:
+  # Trigger the workflow on push to main or develop, except tag creation
+  push:
+    branches:
+      - 'main'
+      - 'develop'
+    tags-ignore:
+      - '**'
+    paths-ignore:
+      - "docs/**"
+      - "CHANGELOG.md"
+      - "README.md"
+
+  # Trigger the workflow on pull request
+  pull_request:
+    paths-ignore:
+      - "docs/**"
+      - "CHANGELOG.md"
+      - "README.md"
+
+  # Trigger the workflow manually
+  workflow_dispatch: ~
+
+  # Trigger after public PR approved for CI
+  pull_request_target:
+    types: [labeled]
+    paths-ignore:
+      - "docs/**"
+      - "CHANGELOG.md"
+      - "README.md"
+
+jobs:
+  # Run CI including downstream packages on self-hosted runners
+  downstream-ci:
+    name: downstream-ci
+    if: ${{ !github.event.pull_request.head.repo.fork && github.event.action != 'labeled' || github.event.label.name == 'approved-for-ci' }}
+    uses: ecmwf/downstream-ci/.github/workflows/downstream-ci.yml@main
+    with:
+      anemoi-utils: ecmwf/anemoi-utils@${{ github.event.pull_request.head.sha || github.sha }}
+      codecov_upload: true
+      # Only run on fedora
+      skip_matrix_jobs: |
+        gnu@debian-11
+        gnu@rocky-8.6
+        clang@rocky-8.6
+        gnu@ubuntu-22.04
+    secrets: inherit
+
+  # # Build downstream packages on HPC
+  # downstream-ci-hpc:
+  #   name: downstream-ci-hpc
+  #   if: ${{ !github.event.pull_request.head.repo.fork && github.event.action != 'labeled' || github.event.label.name == 'approved-for-ci' }}
+  #   uses: ecmwf/downstream-ci/.github/workflows/downstream-ci-hpc.yml@main
+  #   with:
+  #     anemoi-utils: ecmwf/anemoi-utils@${{ github.event.pull_request.head.sha || github.sha }}
+  #   secrets: inherit

anemoi_utils-0.4.40/.github/workflows/pr-conventional-commit.yml

@@ -0,0 +1,22 @@
+# This workflow ensures that the PR title follows the Conventional Commit format.
+name: "[PR] Ensure Conventional Commit Title"
+
+on:
+  pull_request_target:
+    types:
+      - opened
+      - edited
+      - synchronize
+      - reopened
+
+permissions:
+  pull-requests: read
+
+jobs:
+  main:
+    name: Validate PR title
+    runs-on: ubuntu-latest
+    steps:
+      - uses: amannn/action-semantic-pull-request@v5
+        env:
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}

anemoi_utils-0.4.40/.github/workflows/pr-label-ats.yml

@@ -0,0 +1,66 @@
+# This workflow checks that the appropriate ATS labels are applied to a pull request:
+# - "ATS Approval Not Needed": pass
+# - "ATS Approved": pass
+# - "ATS Approval Needed": fail
+# - Missing ATS label: fail
+
+name: "[PR] ATS labels"
+
+on:
+  pull_request:
+    types:
+      - opened
+      - edited
+      - reopened
+      - labeled
+      - unlabeled
+      - synchronize
+
+permissions:
+  pull-requests: read
+
+jobs:
+  check:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Get PR details
+        uses: actions/github-script@v7
+        id: check-pr
+        with:
+          script: |
+            const pr = await github.rest.pulls.get({
+              owner: context.repo.owner,
+              repo: context.repo.repo,
+              pull_number: context.payload.pull_request.number,
+            });
+            const labels = pr.data.labels.map(label => label.name);
+            core.setOutput('labels', JSON.stringify(labels));
+            core.setOutput('author', context.payload.pull_request.user.login);
+
+      - name: Evaluate ATS labels
+        run: |
+          AUTHOR='${{ steps.check-pr.outputs.author }}'
+          if [ "$AUTHOR" == "DeployDuck" ] || [ "$AUTHOR" == "pre-commit-ci[bot]" ]; then
+            echo "Bot PR, skipping."
+            exit 0
+          fi
+          LABELS=$(echo '${{ steps.check-pr.outputs.labels }}' | jq -r '.[]')
+          echo "Labels found:"
+          echo -e "$LABELS\n"
+          echo "Result:"
+          if echo "$LABELS" | grep -qi "ATS approval not needed"; then
+            echo "ATS approval not needed. Passing."
+            EXIT_CODE=0
+          elif echo "$LABELS" | grep -qi "ATS approved"; then
+            echo "ATS Approved. Passing."
+            EXIT_CODE=0
+          elif echo "$LABELS" | grep -qi "ATS approval needed"; then
+            echo "ATS Approval Needed. Failing."
+            EXIT_CODE=1
+          else
+            echo "No ATS approval labels found. Please assign the appropriate ATS label. Failing."
+            EXIT_CODE=1
+          fi
+          echo -e "\nFor more information on ATS labels, see:"
+          echo "https://anemoi.readthedocs.io/en/latest/contributing/guidelines.html#labelling-guidelines"
+          exit $EXIT_CODE

anemoi_utils-0.4.40/.github/workflows/pr-label-conventional-commits.yml

@@ -0,0 +1,48 @@
+# This workflow assigns labels to a pull request based on the Conventional Commits format.
+# This is necessary for release-please to work properly.
+name: "[PR] Label Conventional Commits"
+
+on:
+  pull_request:
+    branches: [main]
+    types:
+      [opened, reopened, labeled, unlabeled]
+
+permissions:
+  pull-requests: write
+
+jobs:
+  assign-labels:
+    runs-on: ubuntu-latest
+    name: Assign labels in pull request
+    if: github.event.pull_request.merged == false
+    steps:
+      - uses: actions/checkout@v3
+      - name: Assign labels from Conventional Commits
+        id: action-assign-labels
+        uses: mauroalderete/action-assign-labels@v1
+        with:
+          pull-request-number: ${{ github.event.pull_request.number }}
+          github-token: ${{ secrets.GITHUB_TOKEN }}
+          conventional-commits: |
+            conventional-commits:
+              - type: 'fix'
+                nouns: ['FIX', 'Fix', 'fix', 'FIXED', 'Fixed', 'fixed']
+                labels: ['bug']
+              - type: 'feature'
+                nouns: ['FEATURE', 'Feature', 'feature', 'FEAT', 'Feat', 'feat']
+                labels: ['enhancement']
+              - type: 'breaking_change'
+                nouns: ['BREAKING CHANGE', 'BREAKING', 'MAJOR']
+                labels: ['breaking change']
+              - type: 'documentation'
+                nouns: ['doc','docs','docu','document','documentation']
+                labels: ['documentation']
+              - type: 'build'
+                nouns: ['build','rebuild','ci']
+                labels: ['CI/CD']
+              - type: 'config'
+                nouns: ['config', 'conf', 'configuration']
+                labels: ['config']
+          maintain-labels-not-matched: true
+          apply-changes: true

anemoi_utils-0.4.40/.github/workflows/pr-label-file-based.yml

@@ -0,0 +1,17 @@
+# This workflow assigns labels to a pull request based on the files changed in the PR.
+# The labels are defined in the `.github/labels.yml` file.
+name: "[PR] Label File-based"
+on:
+  pull_request_target:
+    types: [opened, synchronize]
+
+permissions:
+  contents: read
+  pull-requests: write
+
+jobs:
+  labeler:
+    runs-on: ubuntu-latest
+    steps:
+    - name: Assign labels from file changes
+      uses: actions/labeler@v5

anemoi_utils-0.4.5/.github/workflows/label-public-pr.yml → anemoi_utils-0.4.40/.github/workflows/pr-label-public.yml

@@ -1,5 +1,5 @@
 # Manage labels of pull requests that originate from forks
-name: label-public-pr
+name: "[PR] Label Forks"
 
 on:
   pull_request_target:
@@ -7,4 +7,4 @@ on:
 
 jobs:
   label:
-    uses: ecmwf-actions/reusable-workflows/.github/workflows/label-pr.yml@v2
+    uses: ecmwf/reusable-workflows/.github/workflows/label-pr.yml@v2

anemoi_utils-0.4.40/.github/workflows/python-publish.yml

@@ -0,0 +1,13 @@
+# This workflow will upload a Python Package using Twine when a release is created
+# For more information see: https://help.github.com/en/actions/language-and-framework-guides/using-python-with-github-actions#publishing-to-package-registries
+
+name: Upload Python Package to PyPI
+
+on:
+  release:
+    types: [created]
+
+jobs:
+  deploy:
+    uses: ecmwf/reusable-workflows/.github/workflows/cd-pypi.yml@v2
+    secrets: inherit

anemoi_utils-0.4.40/.github/workflows/python-pull-request.yml

@@ -0,0 +1,25 @@
+# This workflow runs pre-commit checks and pytest tests against multiple platforms and Python versions.
+name: Code Quality and Testing
+
+on:
+  pull_request:
+    types: [opened, synchronize, reopened]
+  push:
+    branches:
+      - main
+  schedule:
+    - cron: "9 2 * * 0" # at 9:02 on sunday
+
+jobs:
+  quality:
+    uses: ecmwf/reusable-workflows/.github/workflows/qa-precommit-run.yml@v2
+    with:
+      skip-hooks: "no-commit-to-branch"
+
+  checks:
+    strategy:
+      matrix:
+        python-version: ["3.10", "3.11", "3.12"]
+    uses: ecmwf/reusable-workflows/.github/workflows/qa-pytest-pyproject.yml@v2
+    with:
+      python-version: ${{ matrix.python-version }}

{anemoi_utils-0.4.5 → anemoi_utils-0.4.40}/.github/workflows/readthedocs-pr-update.yml

@@ -1,4 +1,6 @@
-name: Read the Docs PR Preview
+# This workflow adds a link to the experimental documentation build to the PR.
+# This does NOT trigger a build of the documentation, this is handled through webhooks.
+name: "[PR] Read the Docs Preview"
 on:
   pull_request_target:
     types:

anemoi_utils-0.4.40/.github/workflows/release-please.yml

@@ -0,0 +1,28 @@
+# This workflow uses an action to run Release Please to create a release PR.
+# It is governed by the config and manifest in the root of the repo.
+# For more information see: https://github.com/googleapis/release-please
+name: Run Release Please
+on:
+  push:
+    branches:
+      - main
+      - hotfix/*
+
+permissions:
+  contents: write
+  pull-requests: write
+
+jobs:
+  release-please:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: googleapis/release-please-action@v4
+        with:
+          # this assumes that you have created a personal access token
+          # (PAT) and configured it as a GitHub action secret named
+          # `MY_RELEASE_PLEASE_TOKEN` (this secret name is not important).
+          token: ${{ secrets.RELEASE_PLEASE_TOKEN }}
+          # optional. customize path to .release-please-config.json
+          config-file: .release-please-config.json
+          # Currently releases are done from main
+          target-branch: ${{ github.ref_name }}

anemoi_utils-0.4.40/.gitignore

@@ -0,0 +1,148 @@
+# Python
+__pycache__/
+*.py[cod]
+*$py.class
+*.so
+.Python
+__pypackages__/
+
+# Distribution / packaging
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# Testing and coverage
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Documentation
+docs/_build/
+docs/_api/
+/site
+*.mo
+*.pot
+
+# Environments
+.env
+.envrc
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# IDEs
+.idea/
+.spyderproject
+.spyproject
+.ropeproject
+.vscode/
+*.code-workspace
+*.sublime-project
+*.sublime-workspace
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# Type checking
+.mypy_cache/
+.dmypy.json
+dmypy.json
+.pyre/
+.pytype/
+
+# Data files
+*.grib
+*.grib1
+*.grib2
+*.onnx
+*.ckpt
+*.npy
+*.npz
+*.zarr/
+*.nc
+*.h5
+*.hdf5
+*.pkl
+*.parquet
+*.csv
+*.xlsx
+*.xls
+*.json
+*.txt
+*.zip
+*.db
+*.tgz
+
+# ML artifacts
+wandb/
+mlruns/
+lightning_logs/
+*.ckpt
+*.pt
+*.pth
+runs/
+checkpoints/
+
+# Temporary and system files
+*.swp
+*.download
+*.out
+*.sync
+*.dot
+*.tmp
+*.log
+*.log.*
+.DS_Store
+~*
+tmp/
+temp/
+logs/
+_dev/
+_api/
+./outputs
+*tmp_data/
+
+# Project specific
+?
+?.*
+foo
+bar
+~$images.pptx
+test.py
+test.ipynb
+_version.py
+*.to_upload
+tempCodeRunnerFile.python
+Untitled-*.py
+trace.txt
+?/
+*.prof
+prof/
+*.gz

{anemoi_utils-0.4.5 → anemoi_utils-0.4.40}/.pre-commit-config.yaml

@@ -10,7 +10,7 @@ repos:
     entry: jupyter nbconvert --ClearOutputPreprocessor.enabled=True --inplace
     additional_dependencies: [jupyter]
 - repo: https://github.com/pre-commit/pre-commit-hooks
-  rev: v5.0.0
+  rev: v6.0.0
   hooks:
   - id: check-yaml # Check YAML files for syntax errors only
     args: [--unsafe, --allow-multiple-documents]
@@ -20,37 +20,37 @@ repos:
   - id: no-commit-to-branch # Prevent committing to main / master
   - id: check-added-large-files # Check for large files added to git
   - id: check-merge-conflict # Check for files that contain merge conflict
--   repo: https://github.com/pre-commit/pygrep-hooks
-    rev: v1.10.0  # Use the ref you want to point at
-    hooks:
-    -   id: python-use-type-annotations # Check for missing type annotations
-    -   id: python-check-blanket-noqa # Check for # noqa: all
-    -   id: python-no-log-warn # Check for log.warn
+- repo: https://github.com/pre-commit/pygrep-hooks
+  rev: v1.10.0  # Use the ref you want to point at
+  hooks:
+  - id: python-use-type-annotations # Check for missing type annotations
+  - id: python-check-blanket-noqa # Check for # noqa: all
+  - id: python-no-log-warn # Check for log.warn
 - repo: https://github.com/psf/black-pre-commit-mirror
-  rev: 24.10.0
+  rev: 25.11.0
   hooks:
   - id: black
     args: [--line-length=120]
 - repo: https://github.com/pycqa/isort
-  rev: 5.13.2
+  rev: 7.0.0
   hooks:
   - id: isort
     args:
     - -l 120
     - --force-single-line-imports
     - --profile black
+    - --project anemoi
 - repo: https://github.com/astral-sh/ruff-pre-commit
-  rev: v0.7.2
+  rev: v0.14.7
   hooks:
   - id: ruff
     args:
     - --line-length=120
     - --fix
     - --exit-non-zero-on-fix
-    - --preview
     - --exclude=docs/**/*_.py
 - repo: https://github.com/sphinx-contrib/sphinx-lint
-  rev: v1.0.0
+  rev: v1.0.2
   hooks:
   - id: sphinx-lint
 # For now, we use it. But it does not support a lot of sphinx features
@@ -58,29 +58,28 @@ repos:
   rev: v0.0.14
   hooks:
   - id: rstfmt
-    exclude: 'cli/.*' # Because we use argparse
+    exclude: '(cli|schemas)/.*' # Because we use argparse and pydantic sphinx directives
 - repo: https://github.com/b8raoult/pre-commit-docconvert
   rev: "0.1.5"
   hooks:
   - id: docconvert
     args: ["numpy"]
 - repo: https://github.com/tox-dev/pyproject-fmt
-  rev: "v2.5.0"
+  rev: "v2.11.1"
   hooks:
   - id: pyproject-fmt
--   repo: https://github.com/jshwi/docsig # Check docstrings against function sig
-    rev: v0.64.0
-    hooks:
-    -   id: docsig
-        args:
-        - --ignore-no-params # Allow docstrings without parameters
-        - --check-dunders    # Check dunder methods
-        - --check-overridden # Check overridden methods
-        - --check-protected  # Check protected methods
-        - --check-class      # Check class docstrings
-        - --disable=E113     # Disable empty docstrings
-        - --summary          # Print a summary
-ci:
-  autoupdate_schedule: monthly
+    args: ["--max-supported-python", "3.12"]
+- repo: https://github.com/jshwi/docsig # Check docstrings against function sig
+  rev: v0.71.0
+  hooks:
+  - id: docsig
+    args:
+    - --ignore-no-params # Allow docstrings without parameters
+    - --check-dunders    # Check dunder methods
+    - --check-overridden # Check overridden methods
+    - --check-protected  # Check protected methods
+    - --check-class      # Check class docstrings
+    - --disable=SIG101,SIG102     # Disable empty docstrings
 ci:
   autoupdate_schedule: monthly
+  autoupdate_commit_msg: "chore(deps): pre-commit.ci autoupdate"

anemoi_utils-0.4.40/.readthedocs.yaml

@@ -0,0 +1,19 @@
+version: 2
+
+build:
+  os: ubuntu-22.04
+  tools:
+    python: "3.11"
+  jobs:
+    pre_build:
+    - bash docs/scripts/api_build.sh
+
+sphinx:
+  configuration: docs/conf.py
+
+python:
+  install:
+  - method: pip
+    path: .
+    extra_requirements:
+    - docs