passing more args to context creation cb func

Author: asmirnov69

pandas_flavor/register.py

@@ -29,7 +29,7 @@ def __init__(self, pandas_obj):
             @wraps(method)
             def __call__(self, *args, **kwargs):
                 global cb_create_call_stack_context_manager
-                method_call_obj = cb_create_call_stack_context_manager(method.__name__) if cb_create_call_stack_context_manager else nullcontext()
+                method_call_obj = cb_create_call_stack_context_manager(method.__name__, args, kwargs) if cb_create_call_stack_context_manager else nullcontext()
                 with method_call_obj:
                     if not isinstance(method_call_obj, nullcontext):
                         new_args, new_kwargs = method_call_obj.handle_start_method_call(self._obj, method.__name__, method_signature, args, kwargs)
@@ -64,7 +64,7 @@ def __init__(self, pandas_obj):
             @wraps(method)
             def __call__(self, *args, **kwargs):
                 global cb_create_call_stack_context_manager
-                method_call_obj = cb_create_call_stack_context_manager(method.__name__) if cb_create_call_stack_context_manager else nullcontext()
+                method_call_obj = cb_create_call_stack_context_manager(method.__name__, args, kwargs) if cb_create_call_stack_context_manager else nullcontext()
                 with method_call_obj:
                     if not isinstance(method_call_obj, nullcontext):
                         new_args, new_kwargs = method_call_obj.handle_start_method_call(self._obj, method.__name__, method_signature, args, kwargs)