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]]):
"""
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]
@ -1394,9 +1395,9 @@ class RunnableEach(Serializable, Runnable[List[Input], List[Output]]):
return self.bound.batch(input, config)
async def ainvoke(
self, input: List[Input], config: Optional[RunnableConfig] = None
self, input: List[Input], config: Optional[RunnableConfig] = None, **kwargs: Any
) -> List[Output]:
return await self.bound.abatch(input, config)
return await self.bound.abatch(input, config, **kwargs)
class RunnableBinding(Serializable, Runnable[Input, Output]):

View File

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

View File

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