From fd44ca01d02be150808f13519150eb4f8c171dc2 Mon Sep 17 00:00:00 2001
From: J-Doiron <139803019+J-Doiron@users.noreply.github.com>
Date: Fri, 13 Dec 2024 16:17:40 -0800
Subject: [PATCH] Update shadowofthetombraider.py

Adjusting marker for SOTTR
---
 shadowofthetombraider/shadowofthetombraider.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/shadowofthetombraider/shadowofthetombraider.py b/shadowofthetombraider/shadowofthetombraider.py
index 94fe055..a95e17e 100644
--- a/shadowofthetombraider/shadowofthetombraider.py
+++ b/shadowofthetombraider/shadowofthetombraider.py
@@ -105,7 +105,9 @@ def run_benchmark(keras_service, am):
     elapsed_setup_time = round(time.time() - setup_start_time, 2)
     logging.info("Setup took %f seconds", elapsed_setup_time)
 
-    time.sleep(2)
+    if keras_service.wait_for_word(word="FPS", timeout=30, interval=0.5) is None:
+        logging.info("Did not find the benchmark option on the screen. Did the menu get stuck?")
+        sys.exit(1)
     test_start_time = time.time()
 
     # Wait for benchmark to complete