From 882b97cfd26c1a0a8913ea5dbcb322947e933010 Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Wed, 23 Aug 2023 19:50:20 +0100 Subject: [PATCH] Lint --- libs/langchain/langchain/schema/runnable/base.py | 7 ++++--- libs/langchain/langchain/utilities/openapi.py | 2 +- .../tests/unit_tests/schema/runnable/test_runnable.py | 1 - 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/libs/langchain/langchain/schema/runnable/base.py b/libs/langchain/langchain/schema/runnable/base.py index d9e439fb423..c7dd38865fa 100644 --- a/libs/langchain/langchain/schema/runnable/base.py +++ b/libs/langchain/langchain/schema/runnable/base.py @@ -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]): diff --git a/libs/langchain/langchain/utilities/openapi.py b/libs/langchain/langchain/utilities/openapi.py index 34224a2bbf0..16dd5b680b3 100644 --- a/libs/langchain/langchain/utilities/openapi.py +++ b/libs/langchain/langchain/utilities/openapi.py @@ -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.""" diff --git a/libs/langchain/tests/unit_tests/schema/runnable/test_runnable.py b/libs/langchain/tests/unit_tests/schema/runnable/test_runnable.py index ed40dec28d5..4f3e3b4b910 100644 --- a/libs/langchain/tests/unit_tests/schema/runnable/test_runnable.py +++ b/libs/langchain/tests/unit_tests/schema/runnable/test_runnable.py @@ -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