code: def load_log(fol_path):
      all_paths = filesystem.find_files_with_ext(fol_path, "zlog")
      log_data = {}
      for path in all_paths:
          key = (
              os.path.abspath(path)
              .replace(os.path.abspath(fol_path), "")[1:]
              .replace(".zlog", "")
          )
          log_data[key] = py_io.read_jsonl(path)
      return log_data
  
imports: import os
  import time
  import torch
  import traceback
  from contextlib import contextmanager
  import jiant.utils.python.io
  import jiant.utils.python.filesystem
label_desc: Use image node_id 1 for calling a global function with example usage: load_log(fol_path) and returns: log_data