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

add cleanup code, test=develop (#43305)

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