From 82f0c8e42ce0ba1af22a6b0edb8789aa3b0a9f8a Mon Sep 17 00:00:00 2001 From: iQuxLE Date: Wed, 11 Dec 2024 12:34:02 +0000 Subject: [PATCH] [Fix] Key Error accessing ID from df after hf_download --- src/curategpt/cli.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/curategpt/cli.py b/src/curategpt/cli.py index 53f231f..e9a09ab 100644 --- a/src/curategpt/cli.py +++ b/src/curategpt/cli.py @@ -2549,7 +2549,7 @@ def download_embeddings(path, collection, repo_id, embeddings_filename, metadata df = pd.read_parquet(Path(parquet_download)) store_objects = [ { - "metadata": row.iloc[0], + "metadata": row.iloc[0]['metadata'], "embeddings": row.iloc[1], "document": row.iloc[2] } for _, row in df.iterrows()