code: def reset_tmp_dir():
      os.environ["TRITON_CACHE_DIR"] = tmpdir
      if os.path.exists(tmpdir):
          # https://stackoverflow.com/questions/303200/how-do-i-remove-delete-a-folder-that-is-not-empty
          shutil.rmtree(tmpdir, ignore_errors=True)
  
imports: import importlib.util
  import os
  import shutil
  import tempfile
  import pytest
  import torch
  import triton
  import triton.language
  from triton.runtime.jit import JITFunction
label_desc: Use image node_id 11 for calling a global function with example usage: reset_tmp_dir() without return types