code: def pytorch_wrapper(fun):
      return lambda in1, in2: pytorch_adapter(fun, in1, in2)
  
imports: import numpy
  import nvidia.dali.ops
  import nvidia.dali.types
  import os
  import random
  from functools import partial
  from nvidia.dali.pipeline import Pipeline
  from nvidia.dali import fn, pipeline_def
  from nvidia.dali.python_function_plugin import current_dali_stream
label_desc: Use image node_id 6 for calling a global function with example usage: pytorch_wrapper(fun) and returns: function