[py]: Fix types in edge & firefox options

symonk · symonk · commit e90ad6d36423 · 2022-12-10T18:09:07.000Z
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
