Try to stabilise Safari ci

Author: mcbarton

scripts/automated-notebook-run-script.py

@@ -49,15 +49,29 @@ def wait_for_idle_status(driver, current_cell, start_time, timeout):
 
     print(current_cell.text)
 
+def rewind_to_first_cell(cell):
+    while True:
+        prev = cell.find_elements(
+            By.XPATH,
+            "preceding-sibling::div[contains(@class,'jp-Notebook-cell')][1]"
+        )
+        if not prev:
+            return cell
+        cell = prev[0]
+
+    
 
 def run_notebook(driver, notebook_area, args):
     """This functions runs all the cells of the notebook"""
     print("Running Cells")
     start_time = time.time()
+    current_cell = driver.find_element(
+        By.CSS_SELECTOR, ".jp-Notebook-cell.jp-mod-selected"
+    )
+
+    current_cell = rewind_to_first_cell(current_cell)
+    
     while True:
-        current_cell = driver.find_element(
-            By.CSS_SELECTOR, ".jp-Notebook-cell.jp-mod-selected"
-        )
         editor_divs = current_cell.find_elements(
             By.CSS_SELECTOR, ".jp-InputArea-editor div"
         )
@@ -114,6 +128,11 @@ def run_notebook(driver, notebook_area, args):
             wait_for_idle_status(driver, current_cell, start_time, args.timeout)
 
 
+        current_cell = current_cell.find_elements(
+            By.XPATH,
+            "following-sibling::div[contains(@class,'jp-Notebook-cell')][1]"
+        )
+
 def download_notebook(driver):
     """This function is used to download the notebook currently open."""
     print("Downloading notebook by clicking download button")