diff --git a/spec-0007/transition_to_rng.py b/spec-0007/transition_to_rng.py index de4d069f..99a326c7 100644 --- a/spec-0007/transition_to_rng.py +++ b/spec-0007/transition_to_rng.py @@ -1,5 +1,6 @@ import numpy as np import functools +import inspect import warnings @@ -160,6 +161,17 @@ def wrapper(*args, **kwargs): return fun(*args, **kwargs) + # Add the old parameter name to the function signature + wrapped_signature = inspect.signature(fun) + wrapper.__signature__ = wrapped_signature.replace( + parameters=[ + *wrapped_signature.parameters.values(), + inspect.Parameter( + old_name, inspect.Parameter.KEYWORD_ONLY, default=None + ), + ] + ) + return wrapper return decorator