Update failed_run.py

pushing through the existing function

failed_run.py

@@ -49,8 +49,7 @@ try:
 except FileNotFoundError as e:
     print(f"Could not find {args.run_dir}/error.log")
 
-updated_dataset = Dataset.from_pandas(dataset)
-updated_dataset.push_to_hub("EnergyStarAI/requests_debug", split="test", token=TOKEN)
+
 
 print("Status set to FAILED")
 
@@ -64,6 +63,8 @@ if args.logs_name:
             repo_type="dataset",
         )
     dataset.loc[dataset["model"].isin(args.model_name), ['status']] = "FAILED"
+    updated_dataset = Dataset.from_pandas(dataset)
+    updated_dataset.push_to_hub("EnergyStarAI/requests_debug", split="test", token=TOKEN)
     except Exception as e:
         print("That didn't work. Error:")
         print(e)