Generate stub file for 5.1.1

robotframework · Mar 13, 2021 · 7c660ed · 7c660ed
1 parent 9974127
commit 7c660ed
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/gen_stub.py b/gen_stub.py
@@ -106,8 +106,7 @@ def keyword_line(keyword_arguments, keyword_types, method_name):
 from datetime import timedelta
 from typing import Any, Optional, Union
 
-from selenium.webdriver.remote.webdriver import WebDriver
-from selenium.webdriver.remote.webelement import WebElement
+import selenium
 
 class SeleniumLibrary:
 """

diff --git a/src/SeleniumLibrary/__init__.pyi b/src/SeleniumLibrary/__init__.pyi
@@ -1,8 +1,7 @@
 from datetime import timedelta
 from typing import Any, Optional, Union
 
-from selenium.webdriver.remote.webdriver import WebDriver
-from selenium.webdriver.remote.webelement import WebElement
+import selenium
 
 class SeleniumLibrary:
     def __init__(
@@ -309,7 +308,9 @@ class SeleniumLibrary:
         alias: Optional[str] = None,
         remote_url: Union[bool, str] = False,
         desired_capabilities: Optional[Union[dict, None, str]] = None,
-        ff_profile_dir: Optional[str] = None,
+        ff_profile_dir: Optional[
+            Union[selenium.webdriver.firefox.firefox_profile.FirefoxProfile, str, None]
+        ] = None,
         options: Optional[Any] = None,
         service_log_path: Optional[str] = None,
         executable_path: Optional[str] = None,