code: def write_done(output_dir):
      py_io.write_file("DONE", os.path.join(output_dir, "DONE"))
  
imports: import json
  import numpy
  import os
  import random
  import time
  import torch
  from dataclasses import dataclass
  from typing import Any
  import jiant.utils.python.io
  import jiant.utils.zlog
label_desc: Use image node_id 10 for calling a global function with example usage: write_done(output_dir) without return types