Browse Source

!1199 Remove useless params

From: @liuchongming74
Reviewed-by: @yelihua,@lixiaohui33
Signed-off-by: @lixiaohui33
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 4 years ago
parent
commit
a4acca9ba1
1 changed files with 5 additions and 5 deletions
  1. +5
    -5
      mindinsight/mindconverter/graph_based_converter/framework.py

+ 5
- 5
mindinsight/mindconverter/graph_based_converter/framework.py View File

@@ -147,8 +147,8 @@ def tf_installation_validation(func):
type, inner function. type, inner function.
""" """


def _f(graph_path: str, sample_shape: tuple, output_folder: str, report_folder: str = None,
input_nodes: str = None, output_nodes: str = None):
def _f(graph_path: str, input_nodes: dict, output_nodes: List[str],
output_folder: str, report_folder: str):
not_integral_error = RuntimeIntegrityError( not_integral_error = RuntimeIntegrityError(
f"TensorFlow, " f"TensorFlow, "
f"{get_third_part_lib_validation_error_info(['tf2onnx', 'onnx', 'onnxruntime', 'onnxoptimizer'])} " f"{get_third_part_lib_validation_error_info(['tf2onnx', 'onnx', 'onnxruntime', 'onnxoptimizer'])} "
@@ -171,9 +171,9 @@ def tf_installation_validation(func):
_print_error(not_integral_error) _print_error(not_integral_error)
sys.exit(0) sys.exit(0)


func(graph_path=graph_path, sample_shape=sample_shape,
output_folder=output_folder, report_folder=report_folder,
input_nodes=input_nodes, output_nodes=output_nodes)
func(graph_path=graph_path,
input_nodes=input_nodes, output_nodes=output_nodes,
output_folder=output_folder, report_folder=report_folder)


return _f return _f




Loading…
Cancel
Save