back.pysim: fix completely broken codegen for Switch.

[nmigen.git] / nmigen / tools.py

diff --git a/nmigen/tools.py b/nmigen/tools.py
index 3a2b96871fe6c3f0adbca19e904c1cc4d499ae00..87b3377318aca6c6de4f3939b6fbb3ed07414ff5 100644 (file)
--- a/nmigen/tools.py
+++ b/nmigen/tools.py
@@ -44,11 +44,11 @@ def bits_for(n, require_sign_bit=False):
     return r
 
 
-def deprecated(message):
+def deprecated(message, stacklevel=2):
     def decorator(f):
         @functools.wraps(f)
         def wrapper(*args, **kwargs):
-            warnings.warn(message, DeprecationWarning, stacklevel=2)
+            warnings.warn(message, DeprecationWarning, stacklevel=stacklevel)
             return f(*args, **kwargs)
         return wrapper
     return decorator