Adding handling for space error log in failure handling.

failed_run.py

@@ -23,6 +23,13 @@ parser.add_argument(
     required=True,
     help="Model to benchmark.",
 )
+parser.add_argument(
+    "--error_log",
+    default=None,
+    type=str,
+    required=False,
+    help="Location of space runtime error log -- note this is distinct from an optimum-benchmark log.",
+)
 args = parser.parse_args()
 
 # Updating request
@@ -31,16 +38,31 @@ dataset = load_dataset("EnergyStarAI/requests_debug", split="test", token=TOKEN)
 # Set benchmark to failed
 dataset.loc[dataset["model"].isin(args.model_name), ['status']] = "FAILED"
 
-## Read error message
-# with open(f"{args.run_dir}/error.log", 'r') as file:
-#   error_message = file.read()
-#
-## Add a new column for the error message if necessary
-#if "error_message" not in dataset.columns:
-#    dataset["error_message"] = ""
-#dataset.loc[dataset["model"].isin(args.model_name), ['error_message']] = error_message
+try:
+    # Read error message
+    with open(f"{args.run_dir}/error.log", 'r') as file:
+       error_message = file.read()
+    # Add a new column for the error message if necessary
+    if "error_message" not in dataset.columns:
+        dataset["error_message"] = ""
+    dataset.loc[dataset["model"].isin(args.model_name), ['error_message']] = error_message
+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")
+
+if args.error_log:
+    print("Attempting to save space runtime error log at EnergyStarAI/error_logs")
+    try:
+        api.upload_file(
+            path_or_fileobj=args.error_log,
+            path_in_repo=args.error_log,
+            repo_id="EnergyStarAI/error_logs",
+            repo_type="dataset",
+        )
+    except Exception as e:
+        print("That didn't work. Error:")
+        print(e)