From 4518a8fccbf0cdfdee8ff4ced6ad5c857a00e2dc Mon Sep 17 00:00:00 2001 From: yh_cc Date: Fri, 27 Dec 2019 23:12:29 +0800 Subject: [PATCH] [bug fix] test bug fix --- test/core/test_callbacks.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/core/test_callbacks.py b/test/core/test_callbacks.py index 9a11793f..11cf6704 100644 --- a/test/core/test_callbacks.py +++ b/test/core/test_callbacks.py @@ -134,7 +134,7 @@ class TestCallback(unittest.TestCase): def test_fitlog_callback(self): import fitlog - fitlog.set_log_dir(self.tempdir) + fitlog.set_log_dir(self.tempdir, new_log=True) data_set, model = prepare_env() from fastNLP import Tester tester = Tester(data=data_set, model=model, metrics=AccuracyMetric(pred="predict", target="y")) @@ -164,7 +164,7 @@ class TestCallback(unittest.TestCase): tester = Tester(data=data_set, model=model, metrics=AccuracyMetric(pred="predict", target="y")) import fitlog - fitlog.set_log_dir(self.tempdir) + fitlog.set_log_dir(self.tempdir, new_log=True) tempfile_path = os.path.join(self.tempdir, 'chkt.pt') callbacks = [CheckPointCallback(tempfile_path)]