code: def remove_tmp_path(tmp_path):
      yield
      shutil.rmtree(tmp_path)
  
imports: import shutil
  import pytest
  from tensorflow import keras
label_desc: Use image node_id 2 for calling a global function with example usage: remove_tmp_path(tmp_path) without return types