From 641695a2be0c685ffff83d2084b5e8bdc3402a3a Mon Sep 17 00:00:00 2001
From: 2Shirt <2xShirt@gmail.com>
Date: Thu, 1 Apr 2021 23:22:04 -0600
Subject: [PATCH] Avoid crash if benchmark image fails to generate

---
 scripts/wk/hw/diags.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/scripts/wk/hw/diags.py b/scripts/wk/hw/diags.py
index daba7ab6..cbc83258 100644
--- a/scripts/wk/hw/diags.py
+++ b/scripts/wk/hw/diags.py
@@ -611,7 +611,7 @@ def check_io_benchmark_results(test_obj, rate_list, graph_width, state):
         image_path, state.ost.ticket_id, test_obj.dev.path.name)
       test_obj.upload_report.append(f'Imgur: {imgur_url}')
       test_obj.upload_report.append(f'Nextcloud: {nextcloud_url}')
-    except RuntimeError:
+    except (FileNotFoundError, RuntimeError):
       test_obj.upload_report.append('Failed to upload graph')