From abba07b969a70c2a9d753f0cda28e009cff9b86f Mon Sep 17 00:00:00 2001
From: Asep Haryana Saputra <90584806+MythEclipse@users.noreply.github.com>
Date: Sat, 23 May 2026 10:26:21 +0000
Subject: [PATCH] fix: improve ONNX conversion logging

---
 Machine_Learning/convert_onnx.py | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

diff --git a/Machine_Learning/convert_onnx.py b/Machine_Learning/convert_onnx.py
index b2a0eb2..0ebf7eb 100644
--- a/Machine_Learning/convert_onnx.py
+++ b/Machine_Learning/convert_onnx.py
@@ -2,12 +2,16 @@
 """Convert TensorFlow SavedModel to ONNX format."""
 
 import argparse
+import logging
 import subprocess
 import sys
 from pathlib import Path
 
+logging.basicConfig(level=logging.INFO, format="%(asctime)s [%(levelname)s] %(message)s")
+
 
 def main():
+    """Convert a TensorFlow SavedModel to ONNX format using tf2onnx."""
     parser = argparse.ArgumentParser(
         description="Convert TensorFlow SavedModel to ONNX format"
     )
@@ -32,14 +36,11 @@ def main():
 
     args = parser.parse_args()
 
-    # Check that saved model exists
     if not args.saved_model.exists():
         raise FileNotFoundError(f"SavedModel not found at {args.saved_model}")
 
-    # Create output parent directory if needed
     args.output.parent.mkdir(parents=True, exist_ok=True)
 
-    # Run tf2onnx conversion
     cmd = [
         sys.executable,
         "-m",
@@ -52,9 +53,13 @@ def main():
         str(args.opset),
     ]
 
-    subprocess.run(cmd, check=True)
+    try:
+        subprocess.run(cmd, check=True)
+    except subprocess.CalledProcessError as e:
+        logging.error(f"tf2onnx conversion failed with exit code {e.returncode}")
+        raise
 
-    print(f"ONNX model saved to {args.output}")
+    logging.info(f"ONNX model saved to {args.output}")
 
 
 if __name__ == "__main__":