🔨 Check for successful openvino conversion (#571)

Check for successful openvino conversion
openvinotoolkit · Sep 23, 2022 · de1bea2 · de1bea2
1 parent 5b3fc2b
commit de1bea2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/anomalib/deploy/optimize.py b/anomalib/deploy/optimize.py
@@ -68,7 +68,7 @@ def export_convert(
     if export_mode == "openvino":
         export_path = os.path.join(str(export_path), "openvino")
         optimize_command = "mo --input_model " + str(onnx_path) + " --output_dir " + str(export_path)
-        os.system(optimize_command)
+        assert os.system(optimize_command) == 0, "OpenVINO conversion failed"
         with open(Path(export_path) / "meta_data.json", "w", encoding="utf-8") as metadata_file:
             meta_data = get_model_metadata(model)
             # Convert metadata from torch