fix

1 year ago · 736e0dd46e
parent 5b1812f95b
commit 736e0dd46e
1 changed files with 1 additions and 1 deletions
--- 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