Skip to content

register #

Utilities for registering functions.

Functions:

get_registry_decorator #

get_registry_decorator(
    registry: dict[Any, Any],
) -> Callable[[Any], Callable[[FnTypeVar], FnTypeVar]]

Get a decorator to register functions in a given registry.

Parameters:

  • registry (dict) –

    The registry to register functions in.

Returns:

  • Callable[[Any], Callable[[FnTypeVar], FnTypeVar]] –

    A decorator to register functions in the registry.

Source code in presto/utils/register.py 
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
def get_registry_decorator(
    registry: dict[Any, Any],
) -> Callable[[Any], Callable[[FnTypeVar], FnTypeVar]]:
    """Get a decorator to register functions in a given registry.

    Parameters
    ----------
    registry : dict
        The registry to register functions in.

    Returns
    -------
    Callable[[Any], Callable[[FnTypeVar], FnTypeVar]]
        A decorator to register functions in the registry.
    """

    def register_fn(
        key: Any,
    ) -> Callable[[FnTypeVar], FnTypeVar]:
        """Decorator to register a function in a given registry.

        Parameters
        ----------
        key : Any
            The key to register the function under.

        Returns
        -------
        Callable[[FnTypeVar], FnTypeVar]
            A decorator that registers the function in the registry.
        """

        def decorator(func: FnTypeVar) -> FnTypeVar:
            if key in registry:
                raise ValueError(f"Key {key} is already registered.")
            registry[key] = func
            return func

        return decorator

    return register_fn