Registry

dataphy.visionpack.registry

Registry for vision augmentation transforms.

This module provides a registry system for managing and accessing vision augmentation transforms by name. Transforms can be registered using decorators and retrieved by their registered names.

Classes

AugmenterRegistry

Registry for managing vision augmentation transforms.

This class provides a centralized registry for all available vision augmentation transforms. Transforms can be registered using the @register decorator and retrieved by name using the get() method.

Functions

register(name: str) classmethod

Decorator to register a transform class with the registry.

Parameters:

Name	Type	Description	Default
name	str	The name under which to register the transform	required

Returns:

Type	Description
	Decorator function that registers the transform class

Source code in src/dataphy/visionpack/registry.py

@classmethod
def register(cls, name: str):
    """Decorator to register a transform class with the registry.

    Args:
        name: The name under which to register the transform

    Returns:
        Decorator function that registers the transform class
    """
    def deco(fn: Callable):
        cls._REG[name] = fn
        return fn
    return deco

get(name: str) -> Callable classmethod

Retrieve a registered transform by name.

Parameters:

Name	Type	Description	Default
name	str	The name of the registered transform	required

Returns:

Type	Description
Callable	The transform class constructor

Raises:

Type	Description
KeyError	If the transform name is not registered

Source code in src/dataphy/visionpack/registry.py

@classmethod
def get(cls, name: str) -> Callable:
    """Retrieve a registered transform by name.

    Args:
        name: The name of the registered transform

    Returns:
        The transform class constructor

    Raises:
        KeyError: If the transform name is not registered
    """
    if name not in cls._REG:
        raise KeyError(f"Transform '{name}' not registered")
    return cls._REG[name]