feat: add typehint for v1 driver

This commit is contained in:
Trung Le 2021-04-08 01:53:51 +07:00
parent 91385052ec
commit fbd24a13c0
No known key found for this signature in database
GPG Key ID: 0264396EFC85E502
1 changed files with 2 additions and 2 deletions

View File

@ -37,7 +37,7 @@ TARGET_VERSION = 0
class Chrome:
def __new__(cls, *args, emulate_touch=False, **kwargs):
def __new__(cls, *args, emulate_touch=False, **kwargs) -> _Chrome:
if not ChromeDriverManager.installed:
ChromeDriverManager(*args, **kwargs).install()
@ -105,7 +105,7 @@ class Chrome:
class ChromeOptions:
def __new__(cls, *args, **kwargs):
def __new__(cls, *args, **kwargs) -> _ChromeOptions:
if not ChromeDriverManager.installed:
ChromeDriverManager(*args, **kwargs).install()
if not ChromeDriverManager.selenium_patched: