anemoi-utils 0.4.34__tar.gz → 0.4.36__tar.gz

anemoi_utils-0.4.36/.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.34 → anemoi_utils-0.4.36}/.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]
@@ -41,7 +41,7 @@ repos:
     - --profile black
     - --project anemoi
 - repo: https://github.com/astral-sh/ruff-pre-commit
-  rev: v0.12.7
+  rev: v0.12.11
   hooks:
   - id: ruff
     args:

anemoi_utils-0.4.36/.release-please-manifest.json

@@ -0,0 +1,3 @@
+{
+    ".": "0.4.36"
+}

{anemoi_utils-0.4.34 → anemoi_utils-0.4.36}/CHANGELOG.md

@@ -8,6 +8,30 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 Please add your functional changes to the appropriate section in the PR.
 Keep it human-readable, your future self will thank you!
 
+## [0.4.36](https://github.com/ecmwf/anemoi-utils/compare/0.4.35...0.4.36) (2025-09-22)
+
+
+### Features
+
+* Add aliases to registry ([#219](https://github.com/ecmwf/anemoi-utils/issues/219)) ([37267b5](https://github.com/ecmwf/anemoi-utils/commit/37267b548556a796a01b43abb908011eeec85454))
+* Debug imports ([#182](https://github.com/ecmwf/anemoi-utils/issues/182)) ([1eaa615](https://github.com/ecmwf/anemoi-utils/commit/1eaa61540dc9ac3d5fe82f2c91b7fc98c8bb10af))
+* NoAuth for AML mlflow Logging ([#200](https://github.com/ecmwf/anemoi-utils/issues/200)) ([732182e](https://github.com/ecmwf/anemoi-utils/commit/732182ea5d255ba69ea2ed0a23b307d6f64aaf84))
+* Rich logging ([#209](https://github.com/ecmwf/anemoi-utils/issues/209)) ([3c762a5](https://github.com/ecmwf/anemoi-utils/commit/3c762a593ba2dc734becc54b92984d6dc62967ac))
+* Speedup checkpoint editing - remove compression ([#218](https://github.com/ecmwf/anemoi-utils/issues/218)) ([b49120f](https://github.com/ecmwf/anemoi-utils/commit/b49120f763b0b6ee10c805bab2aa7b973047f963))
+* Use obstore to access s3 buckets ([#210](https://github.com/ecmwf/anemoi-utils/issues/210)) ([da380be](https://github.com/ecmwf/anemoi-utils/commit/da380be71d78274d72bd0a3859ef00b1c80e9469))
+
+
+### Bug Fixes
+
+* Add missing s3 function used by datasets ([#212](https://github.com/ecmwf/anemoi-utils/issues/212)) ([30589e8](https://github.com/ecmwf/anemoi-utils/commit/30589e891fbdb1cff205f0350c63e93a725c7242))
+
+## [0.4.35](https://github.com/ecmwf/anemoi-utils/compare/0.4.34...0.4.35) (2025-08-12)
+
+
+### Bug Fixes
+
+* Config override ([#204](https://github.com/ecmwf/anemoi-utils/issues/204)) ([cdeef1e](https://github.com/ecmwf/anemoi-utils/commit/cdeef1ef95ecd3696fefc751f8a97e90fe357329))
+
 ## [0.4.34](https://github.com/ecmwf/anemoi-utils/compare/0.4.33...0.4.34) (2025-08-11)
 
 

{anemoi_utils-0.4.34/src/anemoi_utils.egg-info → anemoi_utils-0.4.36}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: anemoi-utils
-Version: 0.4.34
+Version: 0.4.36
 Summary: A package to hold various functions to support training of ML models on ECMWF data.
 Author-email: "European Centre for Medium-Range Weather Forecasts (ECMWF)" <software.support@ecmwf.int>
 License:                                  Apache License
@@ -225,6 +225,7 @@ Requires-Python: >=3.10
 License-File: LICENSE
 Requires-Dist: aniso8601
 Requires-Dist: deprecation
+Requires-Dist: entrypoints
 Requires-Dist: importlib-metadata; python_version < "3.10"
 Requires-Dist: multiurl
 Requires-Dist: numpy
@@ -256,7 +257,7 @@ Provides-Extra: provenance
 Requires-Dist: gitpython; extra == "provenance"
 Requires-Dist: nvsmi; extra == "provenance"
 Provides-Extra: s3
-Requires-Dist: boto3>1.36; extra == "s3"
+Requires-Dist: obstore; extra == "s3"
 Provides-Extra: tests
 Requires-Dist: anemoi-utils[mlflow]; extra == "tests"
 Requires-Dist: pytest; extra == "tests"

{anemoi_utils-0.4.34 → anemoi_utils-0.4.36}/pyproject.toml

@@ -41,6 +41,7 @@ dynamic = [ "version" ]
 dependencies = [
   "aniso8601",
   "deprecation",
+  "entrypoints",
   "importlib-metadata; python_version<'3.10'",
   "multiurl",
   "numpy",
@@ -73,7 +74,7 @@ optional-dependencies.mlflow = [ "mlflow-skinny>=2.11.1", "requests" ]
 optional-dependencies.provenance = [ "gitpython", "nvsmi" ]
 
 optional-dependencies.s3 = [
-  "boto3>1.36",
+  "obstore",
 ]
 
 optional-dependencies.tests = [ "anemoi-utils[mlflow]", "pytest", "pytest-mock>=3" ]

{anemoi_utils-0.4.34 → anemoi_utils-0.4.36}/src/anemoi/utils/_environment.py

@@ -17,6 +17,9 @@ class Environment:
     ANEMOI_CONFIG_OVERRIDE_PATH: str
     """Path to the configuration override file for Anemoi."""
 
+    ANEMOI_DEBUG_IMPORTS: bool
+    """Enable debug imports to trace module loading."""
+
     def __setattr__(self, name: str, value: Any) -> None:
         raise AttributeError("Cannot set attributes on Environment class. Use environment variables instead.")
 

{anemoi_utils-0.4.34 → anemoi_utils-0.4.36}/src/anemoi/utils/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.4.34'
-__version_tuple__ = version_tuple = (0, 4, 34)
+__version__ = version = '0.4.36'
+__version_tuple__ = version_tuple = (0, 4, 36)
 
-__commit_id__ = commit_id = 'g1bb37c8aa'
+__commit_id__ = commit_id = 'g67b40f99c'

{anemoi_utils-0.4.34 → anemoi_utils-0.4.36}/src/anemoi/utils/checkpoints.py

@@ -214,7 +214,7 @@ def save_metadata(
             zipf.writestr(entry["path"], value.tobytes())
 
 
-def _edit_metadata(path: str, name: str, callback: Callable, supporting_arrays: dict = None) -> None:
+def _edit_metadata(path: str, name: str, callback: Callable, supporting_arrays: dict | None = None) -> None:
     """Edit metadata in a checkpoint file.
 
     Parameters
@@ -230,41 +230,56 @@ def _edit_metadata(path: str, name: str, callback: Callable, supporting_arrays:
     """
     new_path = f"{path}.anemoi-edit-{time.time()}-{os.getpid()}.tmp"
 
-    found = False
-
-    directory = None
-    with TemporaryDirectory() as temp_dir:
-        zipfile.ZipFile(path, "r").extractall(temp_dir)
-        total = 0
-        for root, dirs, files in os.walk(temp_dir):
-            for f in files:
-                total += 1
-                full = os.path.join(root, f)
-                if f == name:
-                    found = True
-                    callback(full)
-                    directory = os.path.dirname(full)
-
-        if not found:
+    with zipfile.ZipFile(path, "r") as source_zip:
+        file_list = source_zip.namelist()
+
+        # Find the target file and its directory
+        target_file = None
+        directory = None
+        for file_path in file_list:
+            if os.path.basename(file_path) == name:
+                target_file = file_path
+                directory = os.path.dirname(file_path)
+                break
+
+        if target_file is None:
             raise ValueError(f"Could not find '{name}' in {path}")
 
+        # Calculate total files for progress bar
+        total_files = len(file_list)
         if supporting_arrays is not None:
+            total_files += len(supporting_arrays)
+
+        with zipfile.ZipFile(new_path, "w", zipfile.ZIP_STORED) as new_zip:
+            with tqdm.tqdm(total=total_files, desc="Rebuilding checkpoint") as pbar:
+
+                # Copy all files except the target file
+                for file_path in file_list:
+                    if file_path != target_file:
+                        with source_zip.open(file_path) as source_file:
+                            data = source_file.read()
+                            new_zip.writestr(file_path, data)
+                        pbar.update(1)
 
-            for key, entry in supporting_arrays.items():
-                value = entry.tobytes()
-                fname = os.path.join(directory, f"{key}.numpy")
-                os.makedirs(os.path.dirname(fname), exist_ok=True)
-                with open(fname, "wb") as f:
-                    f.write(value)
-                    total += 1
-
-        with zipfile.ZipFile(new_path, "w", zipfile.ZIP_DEFLATED) as zipf:
-            with tqdm.tqdm(total=total, desc="Rebuilding checkpoint") as pbar:
-                for root, dirs, files in os.walk(temp_dir):
-                    for f in files:
-                        full = os.path.join(root, f)
-                        rel = os.path.relpath(full, temp_dir)
-                        zipf.write(full, rel)
+                # Handle the target file with callback
+                with TemporaryDirectory() as temp_dir:
+                    # Extract only the target file
+                    source_zip.extract(target_file, temp_dir)
+                    target_full_path = os.path.join(temp_dir, target_file)
+
+                    # Apply the callback
+                    callback(target_full_path)
+
+                    # Add the modified file to the new zip (if it still exists)
+                    if os.path.exists(target_full_path):
+                        new_zip.write(target_full_path, target_file)
+                    pbar.update(1)
+
+                # Add supporting arrays if provided
+                if supporting_arrays is not None:
+                    for key, entry in supporting_arrays.items():
+                        array_path = os.path.join(directory, f"{key}.numpy") if directory else f"{key}.numpy"
+                        new_zip.writestr(array_path, entry.tobytes())
                         pbar.update(1)
 
     os.rename(new_path, path)

{anemoi_utils-0.4.34 → anemoi_utils-0.4.36}/src/anemoi/utils/cli.py

@@ -16,11 +16,26 @@ import sys
 import traceback
 from collections.abc import Callable
 
+from anemoi.utils import ENV
+
 try:
     import argcomplete
 except ImportError:
     argcomplete = None
 
+
+if ENV.ANEMOI_DEBUG_IMPORTS:
+    from datetime import datetime
+    from importlib.abc import MetaPathFinder
+
+    class ImportTracer(MetaPathFinder):
+        def find_spec(self, fullname, path, target=None):
+            now = datetime.now().isoformat(timespec="milliseconds")
+            print(f"[{now}] Importing {fullname} from {path}")
+            return None  # allow normal import processing to continue
+
+    sys.meta_path.insert(0, ImportTracer())
+
 LOG = logging.getLogger(__name__)
 
 
@@ -29,6 +44,10 @@ class Command:
 
     accept_unknown_args = False
 
+    def check(self, parser: argparse.ArgumentParser, args: argparse.Namespace) -> None:
+        """Check the command arguments."""
+        pass
+
     def run(self, args: argparse.Namespace) -> None:
         """Run the command.
 
@@ -72,6 +91,11 @@ def make_parser(description: str, commands: dict[str, Command]) -> argparse.Argu
         action="store_true",
         help="Debug mode",
     )
+    parser.add_argument(
+        "--rich",
+        action="store_true",
+        help="Use rich for logging",
+    )
 
     subparsers = parser.add_subparsers(help="commands:", dest="command")
     for name, command in commands.items():
@@ -216,16 +240,25 @@ def cli_main(
 
     cmd = commands[args.command]
 
-    logging.basicConfig(
-        format="%(asctime)s %(levelname)s %(message)s",
-        datefmt="%Y-%m-%d %H:%M:%S",
-        level=logging.DEBUG if args.debug else logging.INFO,
-    )
+    if args.rich:
+        from .logs import get_rich_handler
+
+        logging.basicConfig(
+            format="%(message)s", level=logging.DEBUG if args.debug else logging.INFO, handlers=[get_rich_handler()]
+        )
+    else:
+        logging.basicConfig(
+            format="%(asctime)s %(levelname)s %(message)s",
+            datefmt="%Y-%m-%d %H:%M:%S",
+            level=logging.DEBUG if args.debug else logging.INFO,
+        )
 
     if unknown and not cmd.accept_unknown_args:
         # This should trigger an error
         parser.parse_args(test_arguments)
 
+    cmd.check(parser, args)
+
     try:
         if unknown:
             cmd.run(args, unknown)

{anemoi_utils-0.4.34 → anemoi_utils-0.4.36}/src/anemoi/utils/commands/transfer.py

@@ -28,10 +28,14 @@ class Transfer(Command):
             The argument parser to which the arguments will be added.
         """
         command_parser.add_argument(
-            "--source", help="A path to a local file or folder or a URL to a file or a folder on S3."
+            "--source",
+            help="A path to a local file or folder or a URL to a file or a folder on S3.",
+            required=True,
         )
         command_parser.add_argument(
-            "--target", help="A path to a local file or folder or a URL to a file or a folder on S3 or a remote folder."
+            "--target",
+            help="A path to a local file or folder or a URL to a file or a folder on S3 or a remote folder.",
+            required=True,
         )
         command_parser.add_argument(
             "--overwrite",

{anemoi_utils-0.4.34 → anemoi_utils-0.4.36}/src/anemoi/utils/config.py

@@ -15,7 +15,6 @@ import json
 import logging
 import os
 import threading
-import warnings
 from typing import Any
 
 import yaml
@@ -187,7 +186,6 @@ class DotDict(dict):
             The attribute value.
         """
 
-        self.warn_on_mutation(attr)
         value = self.convert_to_nested_dot_dict(value)
         super().__setitem__(attr, value)
 
@@ -201,13 +199,9 @@ class DotDict(dict):
         value : Any
             The value to set.
         """
-        self.warn_on_mutation(key)
         value = self.convert_to_nested_dot_dict(value)
         super().__setitem__(key, value)
 
-    def warn_on_mutation(self, key):
-        warnings.warn("Modifying an instance of DotDict(). This class is intended to be immutable.")
-
     def __repr__(self) -> str:
         """Return a string representation of the DotDict.
 
@@ -486,6 +480,10 @@ def _load_config(
         secret_config = _load_config(secret_name)
         _merge_dicts(config, secret_config)
 
+    if ENV.ANEMOI_CONFIG_OVERRIDE_PATH is not None:
+        override_config = load_any_dict_format(os.path.abspath(ENV.ANEMOI_CONFIG_OVERRIDE_PATH))
+        config = merge_configs(config, override_config)
+
     for env, value in os.environ.items():
 
         if not env.startswith("ANEMOI_CONFIG_"):
@@ -578,9 +576,6 @@ def load_config(
 
     with CONFIG_LOCK:
         config = _load_config(name, secrets, defaults)
-        if ENV.ANEMOI_CONFIG_OVERRIDE_PATH is not None:
-            override_config = _load_config(os.path.abspath(ENV.ANEMOI_CONFIG_OVERRIDE_PATH))
-            merge_configs(config, override_config)
         if CONFIG_PATCH is not None:
             config = CONFIG_PATCH(config)
         return config

{anemoi_utils-0.4.34 → anemoi_utils-0.4.36}/src/anemoi/utils/logs.py

@@ -10,10 +10,10 @@
 
 """Logging utilities."""
 
+import contextvars
 import logging
-import threading
 
-thread_local = threading.local()
+LOGGING_NAME = contextvars.ContextVar("logging_name", default="main")
 
 
 LOGGER = logging.getLogger(__name__)
@@ -27,7 +27,7 @@ def set_logging_name(name: str) -> None:
     name : str
         The name to set for logging.
     """
-    thread_local.logging_name = name
+    LOGGING_NAME.set(name)
 
 
 class ThreadCustomFormatter(logging.Formatter):
@@ -46,7 +46,7 @@ class ThreadCustomFormatter(logging.Formatter):
         str
             The formatted log record.
         """
-        record.logging_name = thread_local.logging_name
+        record.logging_name = LOGGING_NAME.get()
         return super().format(record)
 
 
@@ -58,11 +58,39 @@ def enable_logging_name(name: str = "main") -> None:
     name : str, optional
         The default logging name to set, by default "main".
     """
-    thread_local.logging_name = name
 
-    formatter = ThreadCustomFormatter("%(asctime)s - %(logging_name)s - %(levelname)s - %(message)s")
+    logger = logging.getLogger()
+    is_rich = any(handler.__class__.__name__ == "CustomRichHandler" for handler in logger.handlers)
+
+    set_logging_name(name)
+
+    if is_rich:
+        formatter = ThreadCustomFormatter("%(message)s")
+    else:
+        formatter = ThreadCustomFormatter("%(asctime)s - [%(logging_name)s] - %(levelname)s - %(message)s")
 
     logger = logging.getLogger()
 
     for handler in logger.handlers:
         handler.setFormatter(formatter)
+
+
+def get_rich_handler() -> logging.Handler:
+    """Return a RichHandler with custom formatting for logging."""
+
+    from rich.logging import RichHandler
+    from rich.text import Text
+
+    class CustomRichHandler(RichHandler):
+        def render_message(self, record, message):
+            global width
+
+            text = super().render_message(record, message)
+
+            if hasattr(record, "logging_name"):
+                name = record.logging_name
+                text = Text.assemble(f"[{name}]", (" → ", "dim"), text)
+
+            return text
+
+    return CustomRichHandler(log_time_format="[%X]")

{anemoi_utils-0.4.34 → anemoi_utils-0.4.36}/src/anemoi/utils/mlflow/auth.py

@@ -13,6 +13,8 @@ from __future__ import annotations
 import logging
 import os
 import time
+from abc import ABC
+from abc import abstractmethod
 from datetime import datetime
 from datetime import timezone
 from functools import wraps
@@ -35,7 +37,43 @@ if TYPE_CHECKING:
     from collections.abc import Callable
 
 
-class TokenAuth:
+class AuthBase(ABC):
+    """Base class for authentication implementations."""
+
+    @abstractmethod
+    def __init__(self, *args, **kwargs):
+        pass
+
+    @abstractmethod
+    def save(self, **kwargs):
+        pass
+
+    @abstractmethod
+    def login(self, force_credentials: bool = False, **kwargs):
+        pass
+
+    @abstractmethod
+    def authenticate(self, **kwargs):
+        pass
+
+
+class NoAuth(AuthBase):
+    """No-op authentication class."""
+
+    def __init__(self, *args, **kwargs):
+        self._enabled = False
+
+    def save(self, **kwargs):
+        pass
+
+    def login(self, force_credentials: bool = False, **kwargs):
+        pass
+
+    def authenticate(self, **kwargs):
+        pass
+
+
+class TokenAuth(AuthBase):
     """Manage authentication with a keycloak token server."""
 
     config_file = "mlflow-token.json"

{anemoi_utils-0.4.34 → anemoi_utils-0.4.36}/src/anemoi/utils/registry.py

@@ -97,6 +97,8 @@ class Registry(Generic[T]):
         self.package = package
         self.__registered = {}
         self._sources = {}
+        self._aliases = {}
+        self._warnings = set()
         self.kind = package.split(".")[-1]
         self.key = key
         self.api_version = api_version
@@ -118,7 +120,9 @@ class Registry(Generic[T]):
         """
         return _BY_KIND.get(kind)
 
-    def register(self, name: str, factory: Callable | None = None, source: Any | None = None) -> Wrapper | None:
+    def register(
+        self, name: str, factory: Callable | None = None, source: Any | None = None, aliases: list[str] | None = None
+    ) -> Wrapper | None:
         """Register a factory with the registry.
 
         Parameters
@@ -129,6 +133,8 @@ class Registry(Generic[T]):
             The factory to register, by default None.
         source : Any, optional
             The source of the factory, by default None.
+        aliases : list of str, optional
+            Aliases for the factory, by default None.
 
         Returns
         -------
@@ -136,7 +142,13 @@ class Registry(Generic[T]):
             A wrapper if the factory is None, otherwise None.
         """
 
+        aliases = aliases or []
+
         name = name.replace("_", "-")
+        assert (
+            name not in self._aliases
+        ), f"'{name}' is already registered for '{self._aliases[name]}' in {self.package}"
+        assert name not in aliases, f"'{name}' cannot be an alias for itself in {self.package}"
 
         if factory is None:
             # This happens when the @register decorator is used
@@ -150,6 +162,15 @@ class Registry(Generic[T]):
             warnings.warn(f"Existing: {self._sources[name]}")
             warnings.warn(f"New: {source}")
 
+        for alias in aliases:
+            assert (
+                alias not in self.__registered
+            ), f"Alias '{alias}' is already registered as a factory in {self.package}"
+            alias = alias.replace("_", "-")
+            if alias in self._aliases:
+                warnings.warn(f"Alias '{alias}' is already registered for '{self._aliases[alias]}' in {self.package}")
+            self._aliases[alias] = name
+
         self.__registered[name] = factory
         self._sources[name] = source
 
@@ -188,6 +209,7 @@ class Registry(Generic[T]):
         """
 
         name = name.replace("_", "-")
+        name = self._unalias(name)
 
         ok = name in self.factories
         if not ok:
@@ -213,6 +235,7 @@ class Registry(Generic[T]):
         """
 
         name = name.replace("_", "-")
+        name = self._unalias(name)
 
         if return_none:
             return self.factories.get(name)
@@ -305,6 +328,7 @@ class Registry(Generic[T]):
         """
 
         name = name.replace("_", "-")
+        name = self._unalias(name)
 
         factory = self.lookup(name)
         return factory(*args, **kwargs)
@@ -352,3 +376,33 @@ class Registry(Generic[T]):
         raise ValueError(
             f"Entry '{config}' must either be a string, a dictionary with a single entry, or a dictionary with a '{self.key}' key"
         )
+
+    def _unalias(self, name: str) -> str:
+        """Resolve an alias to its canonical name.
+
+        Parameters
+        ----------
+        name : str
+            The name to resolve.
+
+        Returns
+        -------
+        str
+            The canonical name.
+        """
+        canonical = self._aliases.get(name, name)
+        if canonical != name:
+            warnings.warn(
+                f"Alias '{name}' for '{canonical}' in {self.package} is deprecated and will be removed in a future version.",
+                category=DeprecationWarning,
+                # stacklevel=2,
+            )
+
+        return canonical
+
+    def aliases(self):
+        """Get the aliases."""
+        result = {}
+        for alias, name in self._aliases.items():
+            result.setdefault(name, []).append(alias)
+        return result