Update automated-notebook-run-script.py

Author: mcbarton

scripts/automated-notebook-run-script.py

@@ -175,8 +175,7 @@ def main():
     print("Found element:", download_button)
 
     time.sleep(20)
-    driver.execute_script(
-        """
+    click_download_button_script = """
         const el = arguments[0];
         
         // Force element to be visible and focused
@@ -190,9 +189,11 @@ def main():
         composed: true,   // IMPORTANT for shadow DOM
         view: window
         }));
-        });
-        """,
-        download_button,
+        });  
+    """
+
+    download_button = WebDriverWait(driver, 20).until(
+    lambda d: d.execute_script(click_download_button_script,download_button)
     )
 
     time.sleep(20)