formatting

This commit is contained in:
olgavrou 2023-09-04 08:35:29 -04:00
parent 67dc1a9dd2
commit 4e9aecda90

View File

@ -60,6 +60,7 @@ class PickBestFeatureEmbedder(base.Embedder[PickBestEvent]):
if model is None: if model is None:
from sentence_transformers import SentenceTransformer from sentence_transformers import SentenceTransformer
model = SentenceTransformer("all-mpnet-base-v2") model = SentenceTransformer("all-mpnet-base-v2")
self.model = model self.model = model