add wait buffer

HouGroup · Dec 19, 2023 · 467c099 · 467c099
1 parent 127b0f6
commit 467c099
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mdgo/forcefield/crawler.py b/mdgo/forcefield/crawler.py
@@ -110,7 +110,8 @@ def data_from_pdb(self, pdb_dir: str):
             pdb_dir: The path to the input pdb structure file.
         """
         self.web.get("http://traken.chem.yale.edu/ligpargen/")
-        upload_xpath = "/html/body/div[2]/div/div[2]/form/div/input[2]"
+        upload_xpath = '//*[@id="exampleMOLFile"]'
+        time.sleep(1)
         self.wait.until(EC.presence_of_element_located((By.XPATH, upload_xpath)))
         upload = self.web.find_element(By.XPATH, upload_xpath)
         try: