From 0803723e9ec62ca1da590fc1adb7b2acd59cc136 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9D=8E=E9=B8=BF=E7=AB=A0?= Date: Wed, 1 Apr 2020 15:58:39 +0800 Subject: [PATCH] align two return values --- tests/ut/datavisual/processors/test_graph_processor.py | 2 +- tests/utils/log_operations.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/ut/datavisual/processors/test_graph_processor.py b/tests/ut/datavisual/processors/test_graph_processor.py index 1a263571..3c95989d 100644 --- a/tests/ut/datavisual/processors/test_graph_processor.py +++ b/tests/ut/datavisual/processors/test_graph_processor.py @@ -70,7 +70,7 @@ class TestGraphProcessor: self._train_id = log_dir.replace(summary_base_dir, ".") log_operation = LogOperations() - self._temp_path, self._graph_dict = log_operation.generate_log(PluginNameEnum.GRAPH.value, log_dir) + self._temp_path, self._graph_dict, _ = log_operation.generate_log(PluginNameEnum.GRAPH.value, log_dir) self._generated_path.append(summary_base_dir) self._mock_data_manager = data_manager.DataManager([DataLoaderGenerator(summary_base_dir)]) diff --git a/tests/utils/log_operations.py b/tests/utils/log_operations.py index bc4e9e38..cd0d45a4 100644 --- a/tests/utils/log_operations.py +++ b/tests/utils/log_operations.py @@ -157,7 +157,7 @@ class LogOperations: graph_dict = json.load(load_f) graph_dict = log_generator.generate_log(temp_path, graph_dict) - return temp_path, graph_dict + return temp_path, graph_dict, None steps_list = log_settings.get('steps', [1]) tag_name = log_settings.get('tag', 'default_tag')