code: def find_files_with_ext(base_path, ext):
      return find_files(
          base_path=base_path,
          func=lambda filename: filename.endswith(f".{ext}"),
      )
  
file_path: jiant/jiant/utils/python/filesystem.py
imports: import os
  import importlib
  import sys
  from contextlib import contextmanager
label_desc: Use image node_id 2 for calling a global function with example usage: find_files_with_ext(base_path, ext) and returns: find_files