add logger stop

modelscope · Mar 6, 2024 · 2ca6847 · 2ca6847
1 parent 723f007
commit 2ca6847
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/logger_test.py b/tests/logger_test.py
@@ -64,7 +64,9 @@ def test_logger_chat(self) -> None:
 
     def tearDown(self) -> None:
         """Tear down for LoggerTest."""
-        shutil.rmtree(self.run_dir)
+        logger.stop()
+        if os.path.exists(self.run_dir):
+            shutil.rmtree(self.run_dir)
 
 
 if __name__ == "__main__":