code: def tearDown(self) -> None:
      shutil.rmtree(self.test_dir)
      os.chdir(self.old_cwd)
  
file_path: Ax/ax/runners/tests/test_torchx.py
imports: import os
  import shutil
  import tempfile
  from typing import List
  from ax.core import BatchTrial, Experiment, Objective, OptimizationConfig, Parameter, ParameterType, RangeParameter, SearchSpace
  from ax.metrics.torchx import TorchXMetric
  from ax.modelbridge.dispatch_utils import choose_generation_strategy
  from ax.runners.torchx import TorchXRunner
  from ax.service.scheduler import FailureRateExceededError, Scheduler, SchedulerOptions
  from ax.utils.common.constants import Keys
  from ax.utils.common.testutils import TestCase
  from torchx.components import utils
label_desc: Use image node_id 2 for calling the TorchXRunnerTest obj's underlying member method code with example usage: obj.tearDown() without return types