From c43b54adf47c68d633ab69c3a27bd6d1716c91d0 Mon Sep 17 00:00:00 2001 From: Qi Li Date: Fri, 10 Jun 2022 10:13:17 +0800 Subject: [PATCH] add cleanup code, test=develop (#43305) --- .../paddle/fluid/tests/unittests/ipu/test_weight_decay_ipu.py | 3 +++ python/paddle/fluid/tests/unittests/test_load_op_xpu.py | 3 +++ 2 files changed, 6 insertions(+) diff --git a/python/paddle/fluid/tests/unittests/ipu/test_weight_decay_ipu.py b/python/paddle/fluid/tests/unittests/ipu/test_weight_decay_ipu.py index 627a8fedff6..c2fa0e67272 100644 --- a/python/paddle/fluid/tests/unittests/ipu/test_weight_decay_ipu.py +++ b/python/paddle/fluid/tests/unittests/ipu/test_weight_decay_ipu.py @@ -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 diff --git a/python/paddle/fluid/tests/unittests/test_load_op_xpu.py b/python/paddle/fluid/tests/unittests/test_load_op_xpu.py index f11c39b7b65..8d7f65116b6 100644 --- a/python/paddle/fluid/tests/unittests/test_load_op_xpu.py +++ b/python/paddle/fluid/tests/unittests/test_load_op_xpu.py @@ -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() -- GitLab