diff --git a/tools/pot/openvino/tools/pot/graph/graph_utils.py b/tools/pot/openvino/tools/pot/graph/graph_utils.py index a62c302e462..18d09d0d87e 100644 --- a/tools/pot/openvino/tools/pot/graph/graph_utils.py +++ b/tools/pot/openvino/tools/pot/graph/graph_utils.py @@ -67,7 +67,7 @@ def load_graph(model_config, target_device='ANY'): return graph_from_ir -def save_graph(graph: Graph, save_path, model_name=None, rename_results=False): +def save_graph(graph: Graph, save_path, model_name=None): """ Save model as IR in specified path :param graph: NetworkX model to save :param save_path: path to save the model @@ -86,7 +86,7 @@ def save_graph(graph: Graph, save_path, model_name=None, rename_results=False): graph_copy = deepcopy(graph) add_removed_converts(graph_copy) save_restored_graph(graph=graph_copy, path=save_path, meta_data=graph.meta_data, - name=model_name, rename_results=rename_results) + name=model_name) def model_preprocessing(model): diff --git a/tools/pot/tests/utils/check_graph.py b/tools/pot/tests/utils/check_graph.py index 7a62b00adc9..1d53983a99a 100644 --- a/tools/pot/tests/utils/check_graph.py +++ b/tools/pot/tests/utils/check_graph.py @@ -16,7 +16,7 @@ def check_graph(tmp_path, graph, model_name, model_framework, check_weights=Fals model_name = '_'.join([model_name, model_framework]) ir_name_xml = model_name + '.xml' path_to_ir_xml = tmp_path.joinpath(ir_name_xml) - save_graph(graph, tmp_path.as_posix(), model_name, rename_results=False) + save_graph(graph, tmp_path.as_posix(), model_name) path_to_ref_ir_xml = REFERENCE_MODELS_PATH.joinpath(ir_name_xml)