[py]: Fix types in edge & firefox options

SeleniumHQ · Dec 10, 2022 · e90ad6d · e90ad6d
1 parent 13ed8ce
commit e90ad6d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/py/selenium/webdriver/edge/options.py b/py/selenium/webdriver/edge/options.py
@@ -22,7 +22,7 @@
 class Options(ChromiumOptions):
     KEY = "ms:edgeOptions"
 
-    def __init__(self):
+    def __init__(self) -> None:
         super().__init__()
         self._use_webview = False
 

diff --git a/py/selenium/webdriver/firefox/options.py b/py/selenium/webdriver/firefox/options.py
@@ -14,6 +14,7 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
+import typing
 import warnings
 from typing import Union
 
@@ -24,7 +25,7 @@
 
 
 class Log:
-    def __init__(self):
+    def __init__(self) -> None:
         self.level = None
 
     def to_capabilities(self) -> dict:
@@ -36,9 +37,9 @@ def to_capabilities(self) -> dict:
 class Options(ArgOptions):
     KEY = "moz:firefoxOptions"
 
-    def __init__(self):
+    def __init__(self) -> None:
         super().__init__()
-        self._binary: FirefoxBinary = None
+        self._binary: typing.Optional[FirefoxBinary] = None
         self._preferences: dict = {}
         self._profile = None
         self._proxy = None