Warn at the right stacklevel

pyca · May 3, 2014 · 18e036d · 18e036d
1 parent bac18fc
commit 18e036d
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/cryptography/hazmat/primitives/asymmetric/padding.py b/cryptography/hazmat/primitives/asymmetric/padding.py
@@ -38,7 +38,8 @@ def __init__(self, mgf, salt_length=None):
             warnings.warn(
                 "salt_length is deprecated on MGF1 and should be added via the"
                 " PSS constructor.",
-                utils.DeprecatedIn04
+                utils.DeprecatedIn04,
+                stacklevel=2
             )
         else:
             if (not isinstance(salt_length, six.integer_types) and
@@ -80,7 +81,8 @@ def __init__(self, algorithm, salt_length=None):
             warnings.warn(
                 "salt_length is deprecated on MGF1 and should be passed to "
                 "the PSS constructor instead.",
-                utils.DeprecatedIn04
+                utils.DeprecatedIn04,
+                stacklevel=2
             )
             if (not isinstance(salt_length, six.integer_types) and
                     salt_length is not self.MAX_LENGTH):