code: def get_all_files_from_directory(dir_path, ext):
      file_list = []
      for root, dirs, files in os.walk(dir_path):
          for file in files:
              if file.endswith(ext):
                  file_list.append(os.path.join(root, file))
      return file_list
  
file_path: DALI/dali/test/python/jax_plugin/utils.py
imports: import os
  import numpy
  from nvidia.dali import pipeline_def
  from nvidia.dali.backend import TensorGPU
  import nvidia.dali.fn
  import nvidia.dali.types
label_desc: Use image node_id 4 for calling a global function with example usage: get_all_files_from_directory(dir_path, ext) and returns: file_list