declare-lab/tango2

1import inspect2import warnings3from typing import Any, Dict, Optional, Union4 5from packaging import version6 7 8def deprecate(*args, take_from: Optional[Union[Dict, Any]] = None, standard_warn=True):9    from .. import __version__10 11    deprecated_kwargs = take_from12    values = ()13    if not isinstance(args[0], tuple):14        args = (args,)15 16    for attribute, version_name, message in args:17        if version.parse(version.parse(__version__).base_version) >= version.parse(version_name):18            raise ValueError(19                f"The deprecation tuple {(attribute, version_name, message)} should be removed since diffusers'"20                f" version {__version__} is >= {version_name}"21            )22 23        warning = None24        if isinstance(deprecated_kwargs, dict) and attribute in deprecated_kwargs:25            values += (deprecated_kwargs.pop(attribute),)26            warning = f"The `{attribute}` argument is deprecated and will be removed in version {version_name}."27        elif hasattr(deprecated_kwargs, attribute):28            values += (getattr(deprecated_kwargs, attribute),)29            warning = f"The `{attribute}` attribute is deprecated and will be removed in version {version_name}."30        elif deprecated_kwargs is None:31            warning = f"`{attribute}` is deprecated and will be removed in version {version_name}."32 33        if warning is not None:34            warning = warning + " " if standard_warn else ""35            warnings.warn(warning + message, FutureWarning, stacklevel=2)36 37    if isinstance(deprecated_kwargs, dict) and len(deprecated_kwargs) > 0:38        call_frame = inspect.getouterframes(inspect.currentframe())[1]39        filename = call_frame.filename40        line_number = call_frame.lineno41        function = call_frame.function42        key, value = next(iter(deprecated_kwargs.items()))43        raise TypeError(f"{function} in {filename} line {line_number-1} got an unexpected keyword argument `{key}`")44 45    if len(values) == 0:46        return47    elif len(values) == 1:48        return values[0]49    return values50