fix get_embedding in class AzureSim, passing self.emb_call instead of… (

#239) * fix get_embedding in class AzureSim, passing self.emb_call instead of the client * black reformatted sim.py
landing-ai · Sep 27, 2024 · 6fcdd12 · 6fcdd12
1 parent f0fbff7
commit 6fcdd12
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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(client, x))
+ self.df["embs"] = self.df[sim_key].apply(
+ lambda x: get_embedding(self.emb_call, x)
+ )
 
 
 class OllamaSim(Sim):