bugfix: Fix embed_queries() KeyError

#1
by kmclester - opened
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ def start_haystack():
21
  def load_data_to_store(document_store, retriever):
22
  df = pd.read_csv('monopoly_qa-v1.csv')
23
  questions = list(df.Question)
24
- df['embedding'] = retriever.embed_queries(texts=questions)
25
  df = df.rename(columns={"Question":"content","Answer":"answer"})
26
  df.drop('link to source (to prevent duplicate sources)',axis=1, inplace=True)
27
 
 
21
  def load_data_to_store(document_store, retriever):
22
  df = pd.read_csv('monopoly_qa-v1.csv')
23
  questions = list(df.Question)
24
+ df['embedding'] = retriever.embed_queries(queries=questions).tolist()
25
  df = df.rename(columns={"Question":"content","Answer":"answer"})
26
  df.drop('link to source (to prevent duplicate sources)',axis=1, inplace=True)
27