Merge pull request #1 from iQuxLE/iQuxLE-patch-hf_download

refactor hf_download
iQuxLE · Feb 6, 2025 · d823f00 · d823f00
2 parents 3763ae9 + f7f62c6
commit d823f00
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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'],
+                    "metadata": row.iloc[0],
                     "embeddings": row.iloc[1],
                     "document": row.iloc[2]
                 } for _, row in df.iterrows()