sasha HF Staff commited on
Commit
ba4adc5
·
verified ·
1 Parent(s): 3d75255

Update failed_run.py

Browse files
Files changed (1) hide show
  1. failed_run.py +3 -1
failed_run.py CHANGED
@@ -54,6 +54,8 @@ try:
54
  dataset.loc[dataset["model"].isin([args.model_name]), ['status']] = "COMPLETED"
55
  print("Status set to COMPLETE")
56
  # Add a new column for the error message if necessary
57
-
 
 
58
  updated_dataset = Dataset.from_pandas(dataset)
59
  updated_dataset.push_to_hub("EnergyStarAI/requests_debug", split="test", token=TOKEN)
 
54
  dataset.loc[dataset["model"].isin([args.model_name]), ['status']] = "COMPLETED"
55
  print("Status set to COMPLETE")
56
  # Add a new column for the error message if necessary
57
+ except FileNotFoundError as e:
58
+ print(f"Could not find {args.run_dir}/error.log")
59
+
60
  updated_dataset = Dataset.from_pandas(dataset)
61
  updated_dataset.push_to_hub("EnergyStarAI/requests_debug", split="test", token=TOKEN)