Try to fix previous commit

Author: mcbarton

scripts/automated-notebook-run-script.py

@@ -140,12 +140,10 @@ def run_notebook(driver, notebook_area, args):
                 # happens, but this section is here to try and detect these occurances, and rerun the notebook
                 # from the start if this occurs.
                 print("Parsing failed detected. Restarting notebook run.")
-                restart_button = wait.until(
-                    EC.element_to_be_clickable((
-                    By.CSS_SELECTOR,
-                    "div.lm-Widget.jp-CommandToolbarButton.jp-Toolbar-item[data-jp-item-name='restart'] jp-button"
-                    ))
+                restart_button = driver.find_element(
+                    By.CSS_SELECTOR, "div.lm-Widget.jp-CommandToolbarButton.jp-Toolbar-item[data-jp-item-name='restart'] jp-button"
                 )
+                time.sleep(0.25)
                 restart_button.click()
                 ActionChains(driver).send_keys(Keys.ENTER).perform()
                 current_cell = rewind_to_first_cell(current_cell)