Skip to content

Commit

Permalink
correct driver typing in Session
Browse files Browse the repository at this point in the history
  • Loading branch information
bmos committed Mar 1, 2024
1 parent df3db34 commit 2e3e60f
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions requestium/requestium_session.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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 = {}
Expand Down

0 comments on commit 2e3e60f

Please sign in to comment.