code: def teardown_module(module):
      # clean up: delete the .wgit directory created during this Test
      # Making sure the current directory is ./experimental before removing test dir
      if (Path.cwd().parent.name == "experimental") and (
          Path.cwd().name == "wgit_testing"
      ):
          os.chdir(Path.cwd().parent)
          shutil.rmtree("./wgit_testing/")
      else:
          raise Exception("Exception in testing directory tear down!")
  
file_path: fairscale/tests/experimental/wgit/test_api.py
imports: import json
  import os
  from pathlib import Path
  import random
  import shutil
  import pytest
  import torch
  from torch import nn
  from fairscale.experimental.wgit.repo import Repo, RepoStatus
label_desc: Use image node_id 10 for calling a global function with example usage: teardown_module(module) without return types