Skip to content

Commit

Permalink
black reformatted sim.py
Browse files Browse the repository at this point in the history
  • Loading branch information
hieutrtr committed Sep 27, 2024
1 parent 97b4062 commit efd3440
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion vision_agent/utils/sim.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down

0 comments on commit efd3440

Please sign in to comment.