code: def kms_factory(kms_connection_configuration):
      return InMemoryKmsClient(kms_connection_configuration)
  
file_path: arrow/python/examples/dataset/write_dataset_encrypted.py
imports: import shutil
  import os
  from datetime import timedelta
  import pyarrow
  import pyarrow.dataset
  import pyarrow.parquet.encryption
  from pyarrow.tests.parquet.encryption import InMemoryKmsClient
label_desc: Use image node_id 1 for calling a global function with example usage: kms_factory(kms_connection_configuration) and returns: InMemoryKmsClient