From efd3440580fa38dd94ba1507234df7293e0664dd Mon Sep 17 00:00:00 2001 From: Hieu Tran Date: Fri, 27 Sep 2024 10:57:42 +0700 Subject: [PATCH] black reformatted sim.py --- vision_agent/utils/sim.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/vision_agent/utils/sim.py b/vision_agent/utils/sim.py index 0f3a8174..8bb9b205 100644 --- a/vision_agent/utils/sim.py +++ b/vision_agent/utils/sim.py @@ -125,7 +125,9 @@ def __init__( raise ValueError("key is required if no column 'embs' is present.") if sim_key is not None: - self.df["embs"] = self.df[sim_key].apply(lambda x: get_embedding(self.emb_call, x)) + self.df["embs"] = self.df[sim_key].apply( + lambda x: get_embedding(self.emb_call, x) + ) class OllamaSim(Sim):