clean up

mlcommons · Apr 3, 2024 · 3ba9d49 · 3ba9d49
1 parent d3285f8
commit 3ba9d49
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 12 deletions.
diff --git a/cm-mlops/script/test-mlperf-loadgen-py/customize.py b/cm-mlops/script/test-mlperf-loadgen-py/customize.py
@@ -7,6 +7,20 @@ def preprocess(i):
     env = i['env']
     meta = i['meta']
 
+    print ("********************************************************")
+    print ('- Importing CM library ...')
+    import cmind as cm
+    print ('  SUCCESS!')
+
+    print ('')
+    print ('- List CM repos ...')
+    print ('')
+    r = cm.access({'action':'ls', 'automation':'repo', 'out':'con'})
+    print ('')
+    print ('  SUCCESS!')
+    print ("********************************************************")
+
+
     return {'return':0}
 
 

diff --git a/cm-mlops/script/test-mlperf-loadgen-py/python/main.py b/cm-mlops/script/test-mlperf-loadgen-py/python/main.py
@@ -4,19 +4,7 @@
 
 import numpy as np
 
-print ('')
 print ('- Importing MLPerf loadgen library ...')
 import mlperf_loadgen as lg
 print ('  SUCCESS!')
 
-print ('')
-print ('- Importing CM library ...')
-import cmind as cm
-print ('  SUCCESS!')
-
-print ('')
-print ('- List CM repos ...')
-print ('')
-r = cm.access({'action':'ls', 'automation':'repo', 'out':'con'})
-print ('')
-print ('  SUCCESS!')