Skip to content
2 changes: 2 additions & 0 deletions appium/webdriver/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,5 +16,7 @@
Appium Python Client: WebDriver module
"""

__all__ = ["Remote", "WebElement"]
Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This change looks reasonable to me. This fixes mypy --strict also.


from .webdriver import WebDriver as Remote
from .webelement import WebElement
5 changes: 3 additions & 2 deletions appium/webdriver/webdriver.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@

from typing import TYPE_CHECKING, Any, Callable, Dict, List, Optional, Set, Tuple, Type, Union

from selenium import webdriver
from selenium.common.exceptions import (
InvalidArgumentException,
SessionNotCreatedException,
Expand All @@ -23,6 +22,8 @@
)
from selenium.webdriver.remote.command import Command as RemoteCommand
from selenium.webdriver.remote.remote_connection import RemoteConnection
# `selenium.webdriver.Remote` could be used instead, but Pyright wouldn't locate the class properly.
from selenium.webdriver.remote.webdriver import WebDriver as Remote
from typing_extensions import Self

from appium.common.logger import logger
Expand Down Expand Up @@ -208,7 +209,7 @@ def _get_remote_connection_and_client_config(


class WebDriver(
webdriver.Remote,
Remote,
ActionHelpers,
Activities,
Applications,
Expand Down
Loading