fix(build): resolve mypyc signature generation error in portal.py (#296)

Fixes the mypyc build failure in v0.34.0 by refactoring the `call()` method signatures in `portal.py`.

Author: cofin

sqlspec/utils/portal.py

@@ -149,9 +149,7 @@ async def _async_caller(
         result: _R = await func(*args, **kwargs)
         return result
 
-    def call(
-        self, func: "Callable[..., Coroutine[Any, Any, _R]]", *args: Any, timeout: float = 300.0, **kwargs: Any
-    ) -> _R:
+    def call(self, func: "Callable[..., Coroutine[Any, Any, _R]]", *args: Any, **kwargs: Any) -> _R:
         """Call an async function from synchronous context.
 
         Executes the async function in the background event loop and blocks
@@ -160,8 +158,7 @@ def call(
         Args:
             func: The async function to call.
             *args: Positional arguments to the function.
-            timeout: Maximum seconds to wait for result (default 300).
-            **kwargs: Keyword arguments to the function.
+            **kwargs: Keyword arguments. Supports 'timeout' (float, default 300.0).
 
         Returns:
             Result of the async function.
@@ -170,6 +167,7 @@ def call(
             ImproperConfigurationError: If portal provider not started or timeout reached.
 
         """
+        timeout: float = float(kwargs.pop("timeout", 300.0))
         if self._loop is None or not self.is_running:
             msg = "Portal provider not running. Call start() first."
             raise ImproperConfigurationError(msg)
@@ -233,22 +231,19 @@ def __init__(self, provider: "PortalProvider") -> None:
         """
         self._provider = provider
 
-    def call(
-        self, func: "Callable[..., Coroutine[Any, Any, _R]]", *args: Any, timeout: float = 300.0, **kwargs: Any
-    ) -> _R:
+    def call(self, func: "Callable[..., Coroutine[Any, Any, _R]]", *args: Any, **kwargs: Any) -> _R:
         """Call an async function using the portal provider.
 
         Args:
             func: The async function to call.
             *args: Positional arguments to the function.
-            timeout: Maximum seconds to wait for result (default 300).
-            **kwargs: Keyword arguments to the function.
+            **kwargs: Keyword arguments. Supports 'timeout' (float, default 300.0).
 
         Returns:
             Result of the async function.
 
         """
-        return self._provider.call(func, *args, timeout=timeout, **kwargs)
+        return self._provider.call(func, *args, **kwargs)
 
 
 class PortalManager(metaclass=SingletonMeta):

tests/integration/test_adapters/test_psqlpy/test_extensions/test_events_listen_notify.py

@@ -36,7 +36,7 @@ async def _handler(message: Any) -> None:
 
     try:
         listener = channel.listen("alerts", _handler, poll_interval=0.2)
-        await asyncio.sleep(0.1)
+        await asyncio.sleep(0.3)  # Allow listener to subscribe before publishing
         event_id = await channel.publish("alerts", {"action": "native"})
         for _ in range(200):
             if received:
@@ -82,7 +82,7 @@ async def _handler(message: Any) -> None:
 
     try:
         listener = channel.listen("alerts", _handler, poll_interval=0.2)
-        await asyncio.sleep(0.1)
+        await asyncio.sleep(0.3)  # Allow listener to subscribe before publishing
         event_id = await channel.publish("alerts", {"action": "hybrid"})
         for _ in range(200):
             if received:

tests/integration/test_adapters/test_psycopg/test_extensions/test_events_listen_notify.py

@@ -34,6 +34,7 @@ def test_psycopg_sync_listen_notify(postgres_service: "Any") -> None:
 
     received: list[Any] = []
     listener = channel.listen("alerts", lambda message: received.append(message), poll_interval=0.2)
+    time.sleep(0.3)  # Allow listener to subscribe before publishing
     event_id = channel.publish("alerts", {"action": "ping"})
     for _ in range(200):
         if received:
@@ -67,6 +68,7 @@ async def _handler(message: Any) -> None:
         received.append(message)
 
     listener = channel.listen("alerts", _handler, poll_interval=0.2)
+    await asyncio.sleep(0.3)  # Allow listener to subscribe before publishing
     event_id = await channel.publish("alerts", {"action": "async"})
     for _ in range(200):
         if received:
@@ -101,6 +103,7 @@ def test_psycopg_sync_hybrid_listen_notify_durable(postgres_service: "Any", tmp_
 
     received: list[Any] = []
     listener = channel.listen("alerts", lambda message: received.append(message), poll_interval=0.2)
+    time.sleep(0.3)  # Allow listener to subscribe before publishing
     event_id = channel.publish("alerts", {"action": "hybrid"})
     for _ in range(200):
         if received:
@@ -140,6 +143,7 @@ async def _handler(message: Any) -> None:
         received.append(message)
 
     listener = channel.listen("alerts", _handler, poll_interval=0.2)
+    await asyncio.sleep(0.3)  # Allow listener to subscribe before publishing
     event_id = await channel.publish("alerts", {"action": "hybrid-async"})
     for _ in range(200):
         if received: