Fixes to mlperf readme

Ailurus1 · Feb 5, 2023 · b29e0f6 · b29e0f6
1 parent 2b0b30c
commit b29e0f6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/cm-mlops/script/app-mlperf-inference-nvidia/customize.py b/cm-mlops/script/app-mlperf-inference-nvidia/customize.py
@@ -86,7 +86,7 @@ def preprocess(i):
 def postprocess(i):
 
     env = i['env']
-    if env.get('CM_MLPERF_README', 'no') == "yes":
+    if env.get('CM_MLPERF_README'):
         import cmind as cm
         inp = i['input']
         state = i['state']

diff --git a/cm-mlops/script/app-mlperf-inference-reference/customize.py b/cm-mlops/script/app-mlperf-inference-reference/customize.py
@@ -230,7 +230,7 @@ def postprocess(i):
 
     env = i['env']
 
-    if env.get('CM_MLPERF_README', 'no') == "yes":
+    if env.get('CM_MLPERF_README'):
         import cmind as cm
         inp = i['input']
         state = i['state']