From 5b1812f95b226dba2e88279ac334b4bf247a250a Mon Sep 17 00:00:00 2001 From: olgavrou Date: Mon, 4 Sep 2023 18:35:59 -0400 Subject: [PATCH] fix linting checks --- .../langchain/chains/rl_chain/pick_best_chain.py | 8 ++++---- 1 file changed, 4 insertions(+), 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 31f3cece49c..f048e301320 100644 --- a/libs/langchain/langchain/chains/rl_chain/pick_best_chain.py +++ b/libs/langchain/langchain/chains/rl_chain/pick_best_chain.py @@ -224,17 +224,17 @@ class PickBestFeatureEmbedder(base.Embedder[PickBestEvent]): class PickBestRandomPolicy(base.Policy): - def __init__(self, feature_embedder: base.Embedder, **kwargs): + def __init__(self, feature_embedder: base.Embedder, **kwargs: Any): self.feature_embedder = feature_embedder - def predict(self, event: PickBestEvent): + def predict(self, event: PickBestEvent) -> List[Tuple[int, float]]: num_items = len(event.to_select_from) return [(i, 1.0 / num_items) for i in range(num_items)] - def learn(self, event): + def learn(self, event: PickBestEvent) -> None: pass - def log(self, event): + def log(self, event: PickBestEvent) -> None: pass