Spaces:
Runtime error
Runtime error
See last commit -- fixing on another line.
Browse files- failed_run.py +2 -1
failed_run.py
CHANGED
|
@@ -32,6 +32,7 @@ dataset = load_dataset("EnergyStarAI/requests_debug", split="test", token=TOKEN)
|
|
| 32 |
# Set benchmark to failed
|
| 33 |
dataset.loc[dataset["model"].isin([args.model_name]), ['status']] = "FAILED"
|
| 34 |
|
|
|
|
| 35 |
try:
|
| 36 |
# Read error message
|
| 37 |
with open(f"{args.run_dir}/error.log", 'r') as file:
|
|
@@ -39,7 +40,7 @@ try:
|
|
| 39 |
# Add a new column for the error message if necessary
|
| 40 |
if "error_message" not in dataset.columns:
|
| 41 |
dataset["error_message"] = ""
|
| 42 |
-
dataset.loc[dataset["model"].isin(args.model_name), ['error_message']] = error_message
|
| 43 |
except FileNotFoundError as e:
|
| 44 |
print(f"Could not find {args.run_dir}/error.log")
|
| 45 |
|
|
|
|
| 32 |
# Set benchmark to failed
|
| 33 |
dataset.loc[dataset["model"].isin([args.model_name]), ['status']] = "FAILED"
|
| 34 |
|
| 35 |
+
# TODO: This doesn't have to be try-except, we could actually check if the file is there.
|
| 36 |
try:
|
| 37 |
# Read error message
|
| 38 |
with open(f"{args.run_dir}/error.log", 'r') as file:
|
|
|
|
| 40 |
# Add a new column for the error message if necessary
|
| 41 |
if "error_message" not in dataset.columns:
|
| 42 |
dataset["error_message"] = ""
|
| 43 |
+
dataset.loc[dataset["model"].isin([args.model_name]), ['error_message']] = error_message
|
| 44 |
except FileNotFoundError as e:
|
| 45 |
print(f"Could not find {args.run_dir}/error.log")
|
| 46 |
|