[wptrunner] Fix imports from webdriver module.

Differential Revision: https://phabricator.services.mozilla.com/D218717

bugzilla-url: https://bugzilla.mozilla.org/show_bug.cgi?id=1911984
gecko-commit: a542c78a78d7562eab8bc5a2bfe088af2d71bb2c

tools/wptrunner/wptrunner/executors/asyncactions.py

@@ -1,6 +1,6 @@
 # mypy: allow-untyped-defs
 
-from webdriver.bidi.protocol import BidiWindow
+import webdriver
 
 
 class BidiSessionSubscribeAction:
@@ -20,7 +20,7 @@ async def __call__(self, payload):
                 # value is extracted from the serialized object.
                 if isinstance(context, str):
                     contexts.append(context)
-                elif isinstance(context, BidiWindow):
+                elif isinstance(context, webdriver.bidi.protocol.BidiWindow):
                     contexts.append(context.browsing_context)
                 else:
                     raise ValueError("Unexpected context type: %s" % context)

tools/wptrunner/wptrunner/executors/protocol.py

@@ -5,7 +5,6 @@
 
 from abc import ABCMeta, abstractmethod
 from typing import Any, Awaitable, Callable, ClassVar, List, Mapping, Optional, Type
-from webdriver.bidi.modules.script import Target
 
 
 def merge_dicts(target, source):
@@ -391,7 +390,7 @@ class BidiScriptProtocolPart(ProtocolPart):
     async def call_function(
             self,
             function_declaration: str,
-            target: Target,
+            target: Mapping[str, Any],
             arguments: Optional[List[Mapping[str, Any]]] = None
     ) -> Mapping[str, Any]:
         """