From 736e0dd46ef1f46474884dafce407d0b36b150ec Mon Sep 17 00:00:00 2001 From: olgavrou Date: Mon, 4 Sep 2023 18:40:53 -0400 Subject: [PATCH] fix --- libs/langchain/langchain/chains/rl_chain/pick_best_chain.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 f048e301320..0da0780313c 100644 --- a/libs/langchain/langchain/chains/rl_chain/pick_best_chain.py +++ b/libs/langchain/langchain/chains/rl_chain/pick_best_chain.py @@ -223,7 +223,7 @@ class PickBestFeatureEmbedder(base.Embedder[PickBestEvent]): return self.format_auto_embed_off(event) -class PickBestRandomPolicy(base.Policy): +class PickBestRandomPolicy(base.Policy[PickBestEvent]): def __init__(self, feature_embedder: base.Embedder, **kwargs: Any): self.feature_embedder = feature_embedder