This commit is contained in:
olgavrou 2023-09-04 18:40:53 -04:00
parent 5b1812f95b
commit 736e0dd46e

View File

@ -223,7 +223,7 @@ class PickBestFeatureEmbedder(base.Embedder[PickBestEvent]):
return self.format_auto_embed_off(event) 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): def __init__(self, feature_embedder: base.Embedder, **kwargs: Any):
self.feature_embedder = feature_embedder self.feature_embedder = feature_embedder