提交 0803723e 编写于 作者: 李鸿章

align two return values

上级 b779f3d7
......@@ -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)])
......
......@@ -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')
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册