Update failed_run.py

failed_run.py

@@ -29,9 +29,9 @@ args = parser.parse_args()
 # Updating request
 dataset = load_dataset("EnergyStarAI/requests_debug", split="test", token=TOKEN).to_pandas()
 
-def upload_results():
+def upload_results(output_dir):
     models_ran=[]
-    for f in os.scandir('./runs'):
+    for f in os.scandir(output_dir):
       if f.is_dir():
         for s in os.scandir(f):
           if s.is_dir() and s.name not in ['hooks','info','objects','refs','logs']:
@@ -52,7 +52,7 @@ try:
                 print("Status set to FAILED")
             else:
                 dataset.loc[dataset["model"].isin([args.model_name]), ['status']] = "COMPLETED"
-                upload_results()
+                models_ran = upload_results('./runs')
                 dataset.loc[dataset["model"].isin(models_ran), ['status']] = "COMPLETED"
                 print("Status set to COMPLETE")
     # Add a new column for the error message if necessary