Skip to content

Commit b14e0d4

Browse files
committed
fix: only use positional arguments for function_wrapper
1 parent 060872d commit b14e0d4

1 file changed

Lines changed: 18 additions & 13 deletions

File tree

traincheck/instrumentor/tracer.py

Lines changed: 18 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -155,7 +155,12 @@ def function_wrapper(
155155
*args,
156156
**kwargs,
157157
):
158-
"""Instrumentation for Function"""
158+
"""Instrumentation for Function
159+
160+
When using this wrapper, pass in the control parameters as positional arguments, as any kwargs are passed to the original function.
161+
If you used keyword arguments for the control parameters, you may see errors like:
162+
TypeError: function_wrapper() got multiple values for argument 'arg_name'
163+
"""
159164

160165
global DISABLE_WRAPPER
161166
global PROCESS_ID
@@ -409,18 +414,18 @@ def wrapped(*args, **kwargs):
409414
if increment_step:
410415
META_VARS["step"] += 1
411416
return function_wrapper(
412-
original_function=original_function,
413-
original_function_name=original_function_name,
414-
is_bound_method=is_bound_method,
415-
scan_proxy_in_args=scan_proxy_in_args,
416-
dump_stack_trace=dump_stack_trace,
417-
dump_args=dump_args,
418-
dump_args_config=dump_args_config,
419-
dump_ret=dump_ret,
420-
dump_ret_config=dump_ret_config,
421-
trigger_var_dump=trigger_var_dump,
422-
var_dump_config=var_dump_config,
423-
need_unproxy_args_kwargs=need_unproxy_args_kwargs,
417+
original_function,
418+
original_function_name,
419+
is_bound_method,
420+
scan_proxy_in_args,
421+
dump_stack_trace,
422+
dump_args,
423+
dump_args_config,
424+
dump_ret,
425+
dump_ret_config,
426+
trigger_var_dump,
427+
var_dump_config,
428+
need_unproxy_args_kwargs,
424429
*args,
425430
**kwargs,
426431
)

0 commit comments

Comments
 (0)