fix linting error

This commit is contained in:
olgavrou 2023-09-04 08:56:34 -04:00
parent 0f7cde023b
commit e10980d445

View File

@ -67,7 +67,7 @@ class PickBestFeatureEmbedder(base.Embedder[PickBestEvent]):
self.auto_embed = auto_embed self.auto_embed = auto_embed
@staticmethod @staticmethod
def _str(embedding: List[float]): def _str(embedding: List[float]) -> str:
return " ".join([f"{i}:{e}" for i, e in enumerate(embedding)]) return " ".join([f"{i}:{e}" for i, e in enumerate(embedding)])
def get_label(self, event: PickBestEvent) -> tuple: def get_label(self, event: PickBestEvent) -> tuple: