From 8d10a52525e8e85a2b6930be37e44de7f2b9bcbd Mon Sep 17 00:00:00 2001 From: olgavrou Date: Tue, 29 Aug 2023 05:36:45 -0400 Subject: [PATCH] fix linting complaints --- libs/langchain/langchain/chains/rl_chain/pick_best_chain.py | 1 - .../chains/rl_chain/test_pick_best_text_embedder.py | 4 +--- 2 files changed, 1 insertion(+), 4 deletions(-) diff --git a/libs/langchain/langchain/chains/rl_chain/pick_best_chain.py b/libs/langchain/langchain/chains/rl_chain/pick_best_chain.py index 16e8bf598c9..fa7f18f8fb2 100644 --- a/libs/langchain/langchain/chains/rl_chain/pick_best_chain.py +++ b/libs/langchain/langchain/chains/rl_chain/pick_best_chain.py @@ -6,7 +6,6 @@ from typing import Any, Dict, List, Optional, Tuple, Type, Union import langchain.chains.rl_chain.base as base from langchain.base_language import BaseLanguageModel from langchain.callbacks.manager import CallbackManagerForChainRun -from langchain.chains.base import Chain from langchain.chains.llm import LLMChain from langchain.prompts import BasePromptTemplate diff --git a/libs/langchain/tests/unit_tests/chains/rl_chain/test_pick_best_text_embedder.py b/libs/langchain/tests/unit_tests/chains/rl_chain/test_pick_best_text_embedder.py index acc7491c402..c49bacac608 100644 --- a/libs/langchain/tests/unit_tests/chains/rl_chain/test_pick_best_text_embedder.py +++ b/libs/langchain/tests/unit_tests/chains/rl_chain/test_pick_best_text_embedder.py @@ -264,9 +264,7 @@ def test_pickbest_textembedder_more_namespaces_w_full_label_w_partial_emb() -> N @pytest.mark.requires("vowpal_wabbit_next") -def test_pickbest_textembedder_more_namespaces_w_full_label_w_partial_embed_and_keep() -> ( - None -): +def test_pickbest_textembedder_more_namespaces_w_full_label_w_partial_emakeep() -> None: feature_embedder = pick_best_chain.PickBestFeatureEmbedder(model=MockEncoder()) str1 = "0"