code: def get_input_arg_per_sample(input_data, param_cb, rng):
      return [
          [
              param_cb(
                  SampleDesc(rng, None, sample_idx, batch_idx, sample)
              )
              for sample_idx, sample in enumerate(batch)
          ]
          for batch_idx, batch in enumerate(input_data.data)
      ]
  
imports: import os
  import random
  import numpy
  from typing import List, Union, Callable, Optional
  from dataclasses import dataclass, field
  from nvidia.dali import pipeline_def
  import nvidia.dali.fn
  from nvidia.dali import types
  import nvidia.dali.tensors
  from test_utils import get_dali_extra_path, check_batch
label_desc: Use image node_id 10 for calling a global function with example usage: get_input_arg_per_sample(input_data, param_cb, rng) and returns: unknown