Use mlc logger in scripts instead of print statements 1.0

Fix mlcommons#165, Replacement in the initial three files
sujik18 · Feb 2, 2025 · 8969942 · 8969942
1 parent 2ff9354
commit 8969942
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 4 deletions.
diff --git a/script/app-image-classification-onnx-py/customize.py b/script/app-image-classification-onnx-py/customize.py
@@ -1,7 +1,11 @@
 from mlc import utils
 import os
 import shutil
+import logging
 
+# Configure the logger
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
 def preprocess(i):
 
@@ -45,14 +49,14 @@ def postprocess(i):
         with open(fjson, 'w', encoding='utf-8') as f:
             json.dump(data, f, ensure_ascii=False, indent=4)
     except Exception as e:
-        print('CM warning: {}'.format(e))
+        logger.warning('CM warning: {}'.format(e))
 
     try:
         import yaml
         with open(fyaml, 'w', encoding='utf-8') as f:
             yaml.dump(data, f)
     except Exception as e:
-        print('CM warning: {}'.format(e))
+        logger.warning('CM warning: {}'.format(e))
 
     top_classification = data.get('top_classification', '')
 

diff --git a/script/app-image-corner-detection/customize.py b/script/app-image-corner-detection/customize.py
@@ -1,6 +1,10 @@
 from mlc import utils
 import os
+import logging
 
+# Configure the logger
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
 def preprocess(i):
     os_info = i['os_info']
@@ -38,6 +42,6 @@ def preprocess(i):
 def postprocess(i):
 
     env = i['env']
-    print(env['MLC_OUTPUT'] + " generated in " + env['MLC_RUN_DIR'])
+    logger.info(env['MLC_OUTPUT'] + " generated in " + env['MLC_RUN_DIR'])
 
     return {'return': 0}
diff --git a/script/app-mlperf-automotive-mlcommons-python/customize.py b/script/app-mlperf-automotive-mlcommons-python/customize.py
@@ -3,7 +3,11 @@
 import json
 import shutil
 import subprocess
+import logging
 
+# Configure the logger
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
 def preprocess(i):
 
@@ -72,7 +76,7 @@ def preprocess(i):
         env['MLC_MLPERF_LOADGEN_EXTRA_OPTIONS'] += " --count " + \
             env['MLC_MLPERF_LOADGEN_QUERY_COUNT']
 
-    print("Using MLCommons Inference source from '" +
+    logger.info("Using MLCommons Inference source from '" +
           env['MLC_MLPERF_INFERENCE_SOURCE'] + "'")
 
     if 'MLC_MLPERF_CONF' not in env: