@@ -65,13 +65,19 @@ def _accepts_units(convert_x, convert_y):
6565def decorator (func ):
6666def wrapper (* args ,** kwargs ):
6767axes = args [0 ]
68+ # Bind the incoming arguments to the function signature
6869bound_args = inspect .signature (func ).bind (* args ,** kwargs )
70+ # Get the original arguments - these will be modified later
6971arguments = bound_args .arguments
72+ # Check for data kwarg
7073has_data = (('data' in arguments )and
7174 (arguments ['data' ]is not None ))
7275if has_data :
7376data = arguments ['data' ]
7477
78+ # Loop through each argument to be converted, update the axis
79+ # unit info, convert argument, and replace in *arguments* with
80+ # converted values
7581for arg in convert_x :
7682if has_data and arguments [arg ]in data :
7783data_arg = arguments [arg ]
@@ -100,8 +106,10 @@ def wrapper(*args, **kwargs):
100106
101107if has_data :
102108arguments ['data' ]= data
109+ # Update the arguments with converted values
103110bound_args .arguments = arguments
104111
112+ # Give updated values to the original function
105113args = bound_args .args
106114kwargs = bound_args .kwargs
107115return func (* args ,** kwargs )