From fbd24a13c03c345a292ac4d959ce3928eea261d0 Mon Sep 17 00:00:00 2001
From: Trung Le <jackblk4@gmail.com>
Date: Thu, 8 Apr 2021 01:53:51 +0700
Subject: [PATCH] feat: add typehint for v1 driver

---
 undetected_chromedriver/__init__.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/undetected_chromedriver/__init__.py b/undetected_chromedriver/__init__.py
index 5e5d275..01f2a2e 100644
--- a/undetected_chromedriver/__init__.py
+++ b/undetected_chromedriver/__init__.py
@@ -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: