code: def udf_arg_provider():
      udf_arg_provider = Mock(UDFArgProvider)
      udf_arg_provider.provide_input_args.return_value = {
          "input": Mock()
      }
      udf_arg_provider.provide_params_arg.return_value = {
          "params": Mock()
      }
      udf_arg_provider.provide_additional_kwargs.return_value = {
          "kwarg": Mock()
      }
  
      return udf_arg_provider
  
file_path: sagemaker-python-sdk/tests/unit/sagemaker/feature_store/feature_processor/test_udf_wrapper.py
imports: from __future__ import absolute_import
  from typing import Callable
  import pytest
  from mock import Mock
  from sagemaker.feature_store.feature_processor._feature_processor_config import FeatureProcessorConfig
  from sagemaker.feature_store.feature_processor._udf_arg_provider import UDFArgProvider
  from sagemaker.feature_store.feature_processor._udf_output_receiver import UDFOutputReceiver
  from sagemaker.feature_store.feature_processor._udf_wrapper import UDFWrapper
label_desc: Use image node_id 1 for calling a global function with example usage: udf_arg_provider() and returns: udf_arg_provider