wait page load

htz1992213 · Dec 19, 2023 · befd40e · befd40e
1 parent 836e66f
commit befd40e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mdgo/forcefield/crawler.py b/mdgo/forcefield/crawler.py
@@ -158,7 +158,8 @@ def download_data(self, lmp_name: str):
         """
         print("Structure info uploaded. Rendering force field...")
         lmp_xpath = "/html/body/div[2]/div[2]/div[1]/div/div[14]/form/input[1]"
-        jmol_xpath = self.web.find_element(By.XPATH, "/html/body/div[2]/div[2]/div[2]")
+        jmol_xpath = "/html/body/div[2]/div[2]/div[2]"
+        self.wait.until(EC.presence_of_element_located((By.XPATH, jmol_xpath)))
         self.wait.until(EC.presence_of_element_located((By.XPATH, lmp_xpath)))
         self.web.execute_script("arguments[0].remove();", jmol_xpath)
         self.wait.until(EC.element_to_be_clickable((By.XPATH, lmp_xpath)))