code: def get_files(edge_file_or_dir):
      """get files"""
      if os.path.isdir(edge_file_or_dir):
          ret_files = []
          files = glob.glob(os.path.join(edge_file_or_dir, "*"))
          for file_ in files:
              if os.path.isdir(file_):
                  log.info("%s is a directory, not a file" % file_)
              else:
                  ret_files.append(file_)
      elif "*" in edge_file_or_dir:
          ret_files = []
          files = glob.glob(edge_file_or_dir)
          for file_ in files:
              if os.path.isdir(file_):
                  log.info("%s is a directory, not a file" % file_)
              else:
                  ret_files.append(file_)
      else:
          ret_files = [edge_file_or_dir]
      return ret_files
  
imports: import sys
  import datetime
  import os
  import yaml
  import random
  import shutil
  import six
  import warnings
  import glob
  from collections import defaultdict
  from pgl.utils.logger import log
label_desc: Use image node_id 6 for calling a global function with example usage: get_files(edge_file_or_dir) and returns: ret_files