cog.server.http: quote typenames

Signed-off-by: Yorick van Pelt <[email protected]>
replicate · Nov 22, 2023 · 6af971f · 6af971f
1 parent 0cae154
commit 6af971f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/cog/server/http.py b/python/cog/server/http.py
@@ -99,7 +99,7 @@ class PredictionRequest(schema.PredictionRequest.with_types(input_type=InputType
         T = TypeVar("T")
     def limited(f: "Callable[P, Awaitable[T]]") -> "Callable[P, Awaitable[T]]":
         @functools.wraps(f)
-        async def wrapped(*args: P.args, **kwargs: P.kwargs) -> T:
+        async def wrapped(*args: "P.args", **kwargs: "P.kwargs") -> "T":
             async with http_semaphore:
                 return await f(*args, **kwargs)