fix: Missing/incorrect type hints (#1242)

cobrateam · Feb 15, 2024 · a013448 · a013448
1 parent dc3ae86
commit a013448
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/splinter/browser.py b/splinter/browser.py
@@ -78,7 +78,7 @@
     logger.debug(f"Import Warning: {e}")
 
 
-def get_driver(driver, retry_count=3, config=None, *args, **kwargs):
+def get_driver(driver, retry_count: int = 3, config=None, *args, **kwargs):
     """Try to instantiate the driver.
 
     Common selenium errors are caught and a retry attempt occurs.

diff --git a/splinter/driver/webdriver/__init__.py b/splinter/driver/webdriver/__init__.py
@@ -244,9 +244,9 @@ def find_by(
     self,
     finder,
     finder_kwargs=None,
-    original_find: str = None,
-    original_query: str = None,
-    wait_time: int = None,
+    original_find: Optional[str] = None,
+    original_query: Optional[str] = None,
+    wait_time: Optional[int] = None,
 ):
     """Wrapper for finding elements.