Spaces:
Runtime error
Runtime error
derek-thomas
commited on
Commit
·
615535c
1
Parent(s):
f3e3a3c
Adding debug
Browse files- 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)
|