Update customize.py- fix typo (#915)

mlcommons · Aug 28, 2023 · 8180270 · 8180270
2 parents 96297c3 + 5ae91b9
commit 8180270
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cm-mlops/script/get-tensorrt/customize.py b/cm-mlops/script/get-tensorrt/customize.py
@@ -84,9 +84,9 @@ def preprocess(i):
 
     if env.get('CM_TENSORRT_TAR_FILE_PATH','')=='':
         tags = [ "get", "tensorrt" ]
-        if env.get('CM_TENSORRT_REQUIRE_DEV', '') != 'yes'
+        if env.get('CM_TENSORRT_REQUIRE_DEV', '') != 'yes':
             tags.append("_dev")
-        return {'return': 1, 'error': 'Please envoke cm run script ' + " ".join(tags) + " --tar_file={full path to the TensorRT tar file}'}
+        return {'return': 1, 'error': 'Please envoke cm run script ' + " ".join(tags) + ' --tar_file={full path to the TensorRT tar file}'}
 
 
     print ('Untaring file - can take some time ...')