fix: adding selenium typing

appium · Nov 29, 2024 · 710cd5c · 710cd5c
1 parent 2cababd
commit 710cd5c
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/appium/webdriver/extensions/flutter_integration/flutter_finder.py b/appium/webdriver/extensions/flutter_integration/flutter_finder.py
@@ -15,14 +15,15 @@
 # specific language governing permissions and limitations
 # under the License.
 
-from typing import Tuple
+from typing import Tuple, Union
 
 from appium.webdriver.common.appiumby.AppiumBy import By as AppiumBy
-from appium.webdriver.common.appiumby.AppiumBy import ByType
+from appium.webdriver.common.appiumby.AppiumBy import ByType as AppiumByType
+from selenium.webdriver.common.by import ByType as SeleniumByType
 
 
 class FlutterFinder:
-    def __init__(self, using: ByType, value: str) -> None:
+    def __init__(self, using: Union[SeleniumByType, AppiumByType], value: str) -> None:
         self.using = using
         self.value = value