add editor tests to CI (#5812)

openvinotoolkit · May 26, 2021 · 55c73ab · 55c73ab
1 parent f40ad26
commit 55c73ab
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/.ci/azure/linux.yml b/.ci/azure/linux.yml
@@ -105,6 +105,8 @@ jobs:
         -DENABLE_PYTHON=ON
         -DPYTHON_EXECUTABLE=/usr/bin/python3.6
         -DENABLE_TESTS=ON
+        -DNGRAPH_ONNX_IMPORT_ENABLE=ON
+        -DNGRAPH_ONNX_EDITOR_ENABLE=ON
         -DENABLE_FASTER_BUILD=ON
         -DIE_EXTRA_MODULES=$(OPENVINO_CONTRIB_REPO_DIR)/modules
         $(REPO_DIR)

diff --git a/ngraph/frontend/onnx_editor/src/detail/subgraph_extraction.cpp b/ngraph/frontend/onnx_editor/src/detail/subgraph_extraction.cpp
@@ -119,7 +119,8 @@ namespace
         }
         else
         {
-            // If tensor descriptor couldn't be found value info has to be specified as fully dynamic:
+            // If tensor descriptor couldn't be found value info has to be specified
+            // as fully dynamic:
             // - Fully dynamic shape
             // - Unknown data type
             auto dynamic_value_info = ONNX_NAMESPACE::ValueInfoProto();

diff --git a/ngraph/test/onnx/onnx_editor.cpp b/ngraph/test/onnx/onnx_editor.cpp
@@ -1453,7 +1453,7 @@ NGRAPH_TEST(onnx_editor, cut_operator_with_no_schema)
     ONNXModelEditor editor{file_util::path_join(
         SERIALIZED_ZOO, "onnx/model_editor/unknown_input_value_info.prototxt")};
 
-    editor.cut_graph_fragment({{InputEdge(1, "X")}}, {});
+    editor.cut_graph_fragment({{InputEdge{1, 0}}}, {});
 
     const auto ref_model = file_util::path_join(
         SERIALIZED_ZOO, "onnx/model_editor/reference/unknown_input_value_info.prototxt");