derek-thomas commited on
Commit
615535c
·
1 Parent(s): f3e3a3c

Adding debug

Browse files
Files changed (1) hide show
  1. src/utilities.py +1 -0
src/utilities.py CHANGED
@@ -52,6 +52,7 @@ def merge_and_update_datasets(dataset, original_dataset):
52
  # Iterate over the DataFrame rows where 'embedding' is None
53
  for index, row in merged_df[merged_df['embedding'].isnull()].iterrows():
54
  # Update 'embedding' for the current row using our function
 
55
  merged_df.at[index, 'embedding'] = update_embeddings(content=row['content'], client=client)
56
 
57
  dataset['train'] = Dataset.from_pandas(merged_df)
 
52
  # Iterate over the DataFrame rows where 'embedding' is None
53
  for index, row in merged_df[merged_df['embedding'].isnull()].iterrows():
54
  # Update 'embedding' for the current row using our function
55
+ logger.info(f"Updating index: {index}")
56
  merged_df.at[index, 'embedding'] = update_embeddings(content=row['content'], client=client)
57
 
58
  dataset['train'] = Dataset.from_pandas(merged_df)