This commit is contained in:
Nuno Campos 2023-08-23 19:50:20 +01:00
parent 3ddabe8b2c
commit 882b97cfd2
3 changed files with 5 additions and 5 deletions

View File

@ -1369,7 +1369,8 @@ class RunnableLambda(Runnable[Input, Output]):
class RunnableEach(Serializable, Runnable[List[Input], List[Output]]): class RunnableEach(Serializable, Runnable[List[Input], List[Output]]):
""" """
A runnable that delegates calls to another runnable with each element of the input sequence. A runnable that delegates calls to another runnable
with each element of the input sequence.
""" """
bound: Runnable[Input, Output] bound: Runnable[Input, Output]
@ -1394,9 +1395,9 @@ class RunnableEach(Serializable, Runnable[List[Input], List[Output]]):
return self.bound.batch(input, config) return self.bound.batch(input, config)
async def ainvoke( async def ainvoke(
self, input: List[Input], config: Optional[RunnableConfig] = None self, input: List[Input], config: Optional[RunnableConfig] = None, **kwargs: Any
) -> List[Output]: ) -> List[Output]:
return await self.bound.abatch(input, config) return await self.bound.abatch(input, config, **kwargs)
class RunnableBinding(Serializable, Runnable[Input, Output]): class RunnableBinding(Serializable, Runnable[Input, Output]):

View File

@ -54,7 +54,7 @@ if _PYDANTIC_MAJOR_VERSION == 1:
try: try:
from openapi_schema_pydantic import OpenAPI from openapi_schema_pydantic import OpenAPI
except ImportError: except ImportError:
OpenAPI = object OpenAPI = object # type: ignore
class OpenAPISpec(OpenAPI): class OpenAPISpec(OpenAPI):
"""OpenAPI Model that removes mis-formatted parts of the spec.""" """OpenAPI Model that removes mis-formatted parts of the spec."""

View File

@ -1,6 +1,5 @@
from typing import Any, Dict, List, Optional from typing import Any, Dict, List, Optional
from uuid import UUID from uuid import UUID
from xml.dom import ValidationErr
import pytest import pytest
from freezegun import freeze_time from freezegun import freeze_time