From 2e3e60fde5bbf86cd6a115e7eb7823da550f913e Mon Sep 17 00:00:00 2001 From: Wil Thieme Date: Thu, 29 Feb 2024 20:52:54 -0500 Subject: [PATCH] correct driver typing in Session --- requestium/requestium_session.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/requestium/requestium_session.py b/requestium/requestium_session.py index c0308ab..4e87ffd 100644 --- a/requestium/requestium_session.py +++ b/requestium/requestium_session.py @@ -3,8 +3,8 @@ import requests import tldextract -from selenium import webdriver from selenium.webdriver import ChromeService +from selenium.webdriver.remote.webdriver import WebDriver from .requestium_chrome import RequestiumChrome from .requestium_response import RequestiumResponse @@ -25,7 +25,7 @@ class Session(requests.Session): """ def __init__( - self, webdriver_path: str = None, headless: bool = False, default_timeout: float = 5, webdriver_options: dict = None, driver: webdriver = None + self, webdriver_path: str = None, headless: bool = False, default_timeout: float = 5, webdriver_options: dict = None, driver: WebDriver = None ) -> None: if webdriver_options is None: webdriver_options = {}