Skip to content

Commit 5ffefe8

Browse files
patrickvonplatenThomas Capelle
authored andcommitted
[Deprecate] Correct stacklevel (huggingface#1483)
* Correct stacklevel * fix
1 parent 40c0865 commit 5ffefe8

File tree

2 files changed

+7
-1
lines changed

2 files changed

+7
-1
lines changed

src/diffusers/utils/deprecation_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ def deprecate(*args, take_from: Optional[Union[Dict, Any]] = None, standard_warn
3232

3333
if warning is not None:
3434
warning = warning + " " if standard_warn else ""
35-
warnings.warn(warning + message, FutureWarning)
35+
warnings.warn(warning + message, FutureWarning, stacklevel=2)
3636

3737
if isinstance(deprecated_kwargs, dict) and len(deprecated_kwargs) > 0:
3838
call_frame = inspect.getouterframes(inspect.currentframe())[1]

tests/test_utils.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -162,3 +162,9 @@ def test_deprecate_incorrect_no_standard_warn(self):
162162
deprecate(("deprecated_arg", self.higher_version, "This message is better!!!"), standard_warn=False)
163163

164164
assert str(warning.warning) == "This message is better!!!"
165+
166+
def test_deprecate_stacklevel(self):
167+
with self.assertWarns(FutureWarning) as warning:
168+
deprecate(("deprecated_arg", self.higher_version, "This message is better!!!"), standard_warn=False)
169+
assert str(warning.warning) == "This message is better!!!"
170+
assert "diffusers/tests/test_utils.py" in warning.filename

0 commit comments

Comments
 (0)