Browse Source

align two return values

tags/v0.2.0-alpha
李鸿章 5 years ago
parent
commit
0803723e9e
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      tests/ut/datavisual/processors/test_graph_processor.py
  2. +1
    -1
      tests/utils/log_operations.py

+ 1
- 1
tests/ut/datavisual/processors/test_graph_processor.py View File

@@ -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)])


+ 1
- 1
tests/utils/log_operations.py View File

@@ -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')


Loading…
Cancel
Save