From 1dc49750b82c85fcdfd080ec894af7a268a3ae70 Mon Sep 17 00:00:00 2001
From: Joel Maher <jmaher@mozilla.com>
Date: Thu, 23 Jun 2022 12:09:48 +0000
Subject: [PATCH] Bug 1773889 - exit with TBPL_RETRY for failed condprof job.
 r=sparky

Differential Revision: https://phabricator.services.mozilla.com/D150052
---
 testing/condprofile/condprof/main.py | 29 +++++++++++++++-------------
 1 file changed, 16 insertions(+), 13 deletions(-)

diff --git a/testing/condprofile/condprof/main.py b/testing/condprofile/condprof/main.py
index c1d732abdaa6..5794cdd2059a 100644
--- a/testing/condprofile/condprof/main.py
+++ b/testing/condprofile/condprof/main.py
@@ -62,19 +62,22 @@ def main(args=sys.argv[1:]):
 
     from condprof.runner import run  # NOQA
 
-    run(
-        args.archive,
-        args.firefox,
-        args.scenario,
-        args.profile,
-        args.customization,
-        args.visible,
-        args.archives_dir,
-        args.force_new,
-        args.strict,
-        args.geckodriver,
-        args.device_name,
-    )
+    try:
+        run(
+            args.archive,
+            args.firefox,
+            args.scenario,
+            args.profile,
+            args.customization,
+            args.visible,
+            args.archives_dir,
+            args.force_new,
+            args.strict,
+            args.geckodriver,
+            args.device_name,
+        )
+    except Exception:
+        sys.exit(4)
 
 
 if __name__ == "__main__":