Archives/langchain
2
0
Fork 1
mirror of https://github.com/hwchase17/langchain synced 2024-11-04 06:00:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix linting checks

Browse Source
This commit is contained in:
olgavrou 2023-09-04 18:35:59 -04:00
parent f1d144cd6c
commit 5b1812f95b
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
libs/langchain/langchain/chains/rl_chain/pick_best_chain.py
View File

@ -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