update

intelligent-machine-learning · Oct 11, 2024 · d594ee2 · d594ee2
1 parent 3c95822
commit d594ee2
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/dlrover/python/tests/test_diagnosis_agent.py b/dlrover/python/tests/test_diagnosis_agent.py
@@ -51,7 +51,6 @@ def setUp(self):
 
  def tearDown(self):
  os.environ.clear()
- self._master.stop()
 
  def test_diagnose_training(self):
  file = "data/training.log"

diff --git a/dlrover/python/tests/test_inference_chain.py b/dlrover/python/tests/test_inference_chain.py
@@ -53,7 +53,7 @@ def setUp(self):
  MasterClient._instance = build_master_client(self._addr, 1)
 
  def tearDown(self):
- self._master.stop()
+ os.environ.clear()
 
  def test_check_training_hang_operator(self):
  operator = CheckTrainingHangOperator(None)