diff --git a/src/common/utils/selenium.py b/src/common/utils/selenium.py index 9e543d4..7c3d1b0 100644 --- a/src/common/utils/selenium.py +++ b/src/common/utils/selenium.py @@ -2,12 +2,12 @@ import random import string import typing -from selenium.common.exceptions import StaleElementReferenceException, TimeoutException -from selenium.webdriver.chrome.webdriver import WebDriver +from selenium.webdriver.remote.webdriver import WebDriver +from selenium.webdriver.remote.webelement import WebElement from selenium.webdriver.common.by import By from selenium.webdriver.support import expected_conditions as EC from selenium.webdriver.support.wait import WebDriverWait -from selenium.webdriver.remote.webelement import WebElement +from selenium.common.exceptions import StaleElementReferenceException, TimeoutException driver: WebDriver|None = None attempts, timeout, interval = int(), float(), float()