(*args, take_from: dict | Any | None = None, standard_warn=True, stacklevel=2)
| 54 | |
| 55 | |
| 56 | def deprecate(*args, take_from: dict | Any | None = None, standard_warn=True, stacklevel=2): |
| 57 | from .. import __version__ |
| 58 | |
| 59 | deprecated_kwargs = take_from |
| 60 | values = () |
| 61 | if not isinstance(args[0], tuple): |
| 62 | args = (args,) |
| 63 | |
| 64 | for attribute, version_name, message in args: |
| 65 | if version.parse(version.parse(__version__).base_version) >= version.parse(version_name): |
| 66 | raise ValueError( |
| 67 | f"The deprecation tuple {(attribute, version_name, message)} should be removed since diffusers'" |
| 68 | f" version {__version__} is >= {version_name}" |
| 69 | ) |
| 70 | |
| 71 | warning = None |
| 72 | if isinstance(deprecated_kwargs, dict) and attribute in deprecated_kwargs: |
| 73 | values += (deprecated_kwargs.pop(attribute),) |
| 74 | warning = f"The `{attribute}` argument is deprecated and will be removed in version {version_name}." |
| 75 | elif hasattr(deprecated_kwargs, attribute): |
| 76 | values += (getattr(deprecated_kwargs, attribute),) |
| 77 | warning = f"The `{attribute}` attribute is deprecated and will be removed in version {version_name}." |
| 78 | elif deprecated_kwargs is None: |
| 79 | warning = f"`{attribute}` is deprecated and will be removed in version {version_name}." |
| 80 | |
| 81 | if warning is not None: |
| 82 | warning = warning + " " if standard_warn else "" |
| 83 | warnings.warn(warning + message, FutureWarning, stacklevel=stacklevel) |
| 84 | |
| 85 | if isinstance(deprecated_kwargs, dict) and len(deprecated_kwargs) > 0: |
| 86 | call_frame = inspect.getouterframes(inspect.currentframe())[1] |
| 87 | filename = call_frame.filename |
| 88 | line_number = call_frame.lineno |
| 89 | function = call_frame.function |
| 90 | key, value = next(iter(deprecated_kwargs.items())) |
| 91 | raise TypeError(f"{function} in {filename} line {line_number - 1} got an unexpected keyword argument `{key}`") |
| 92 | |
| 93 | if len(values) == 0: |
| 94 | return |
| 95 | elif len(values) == 1: |
| 96 | return values[0] |
| 97 | return values |
searching dependent graphs…