未验证 提交 c43b54ad 编写于 作者: Q Qi Li 提交者: GitHub

add cleanup code, test=develop (#43305)

上级 0d719718
...@@ -35,6 +35,9 @@ class TestBase(IPUOpTest): ...@@ -35,6 +35,9 @@ class TestBase(IPUOpTest):
self.temp_dir = tempfile.TemporaryDirectory() self.temp_dir = tempfile.TemporaryDirectory()
self.model_path = os.path.join(self.temp_dir.name, "weight_decay") self.model_path = os.path.join(self.temp_dir.name, "weight_decay")
def tearDown(self):
self.temp_dir.cleanup()
def set_atol(self): def set_atol(self):
self.atol = 1e-6 self.atol = 1e-6
......
...@@ -51,6 +51,9 @@ class TestLoadOpXpu(unittest.TestCase): ...@@ -51,6 +51,9 @@ class TestLoadOpXpu(unittest.TestCase):
dirname=self.model_path, dirname=self.model_path,
main_program=main_prog) main_program=main_prog)
def tearDown(self):
self.temp_dir.cleanup()
def test_load_xpu(self): def test_load_xpu(self):
main_prog = fluid.Program() main_prog = fluid.Program()
start_prog = fluid.Program() start_prog = fluid.Program()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册