Update failed_run.py

failed_run.py

@@ -46,7 +46,7 @@ try:
     # Read error message
     with open(f"{args.run_dir}/error.log", 'r') as file:
         for f in file.readlines():
-            if 'RuntimeError' in f:
+            if 'Traceback (most recent call last):' in f:
                 error_message = f
                 dataset.loc[dataset["model"].isin([args.model_name]), ['status']] = "FAILED"
                 print("Status set to FAILED")