Update create_results.py

create_results.py

@@ -21,20 +21,7 @@ api.upload_folder(
 requests = load_dataset("EnergyStarAI/requests_debug", split="test", token=TOKEN)
 requests_dset = requests.to_pandas()
 
-models_ran=[]
-for dir, path, files in os.walk(out_dir):
-    print("Dir is: ")
-    print(dir)
-    print("Path is:")
-    print(path)
-    print("Files are:")
-    print(files)
-    for fid in files:
-        print("fid is:")
-        print(fid)
-        model = os.path.join(dir, fid)
-        models_ran.append(model)
-
+models_ran=[ f.path for f in os.scandir(out_dir) if f.is_dir() ]
 print("Models ran are: " + str(models_ran))
 
 requests_dset.loc[requests_dset["model"].isin(models_ran), ['status']] = "COMPLETED"