Merge branch 'main' of https://github.com/htz1992213/mdgo

HouGroup · Dec 19, 2023 · 8f82809 · 8f82809
2 parents b38841a + 137255a
commit 8f82809
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mdgo/forcefield/crawler.py b/mdgo/forcefield/crawler.py
@@ -73,6 +73,7 @@ def __init__(
             "profile.managed_default_content_settings.images": 2,
         }
         self.options = webdriver.ChromeOptions()
+        self.server = webdriver.ChromeService(chromedriver_dir)
         self.options.add_argument(
             'user-agent="Mozilla/5.0 '
             "(Macintosh; Intel Mac OS X 10_14_6) "
@@ -87,7 +88,7 @@ def __init__(
         if chromedriver_dir is None:
             self.web = webdriver.Chrome(options=self.options)
         else:
-            self.web = webdriver.Chrome(chromedriver_dir, options=self.options)
+            self.web = webdriver.Chrome(service=self.server, options=self.options)
         self.wait = WebDriverWait(self.web, 10)
         self.web.get("http://zarbi.chem.yale.edu/ligpargen/")
         time.sleep(1)