| 6 | |
| 7 | |
| 8 | def deprecate(*args, take_from: Optional[Union[Dict, Any]] = None, standard_warn=True, stacklevel=2): |
| 9 | from .. import __version__ |
| 10 | |
| 11 | deprecated_kwargs = take_from |
| 12 | 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 = None |
| 24 | 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=stacklevel) |
| 36 | |
| 37 | if isinstance(deprecated_kwargs, dict) and len(deprecated_kwargs) > 0: |
| 38 | call_frame = inspect.getouterframes(inspect.currentframe())[1] |
| 39 | filename = call_frame.filename |
| 40 | line_number = call_frame.lineno |
| 41 | function = call_frame.function |
| 42 | 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 | return |
| 47 | elif len(values) == 1: |
| 48 | return values[0] |
| 49 | return values |