Merge from CTuning (#1035)

mlcommons · Dec 8, 2023 · 6d9dfda · 6d9dfda
2 parents 50b6ca7 + 575419b
commit 6d9dfda
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 7 deletions.
diff --git a/tests/script/process_dockerfile.py b/tests/script/process_dockerfile.py
@@ -14,9 +14,9 @@
         continue
     script_path = os.path.dirname(file)
     f = open(file)
-    if f.endswith(".json"):
+    if file.endswith(".json"):
         data = json.load(f)
-    elif f.endswith(".yaml"):
+    elif file.endswith(".yaml"):
         data = yaml.safe_load(f)
     uid = data['uid']
 

diff --git a/tests/script/process_readme.py b/tests/script/process_readme.py
@@ -8,16 +8,15 @@
 files=sys.argv[1:]
 
 for file in files:
-    filename = os.path.basename(file)
-    if not filename.endswith("_cm.json") and not filename.endswith("_cm.yaml"):
+    if not file.endswith("_cm.json") and not file.endswith("_cm.yaml"):
         continue
-    if not str(file).startswith(os.path.join("cm-mlops", "script")):
+    if not file.startswith(os.path.join("cm-mlops", "script")):
         continue
     script_path = os.path.dirname(file)
     f = open(file)
-    if f.endswith(".json"):
+    if file.endswith(".json"):
         data = json.load(f)
-    elif f.endswith(".yaml"):
+    elif file.endswith(".yaml"):
         data = yaml.safe_load(f)
     uid = data['uid']