modify writing to parquet

app.py

@@ -83,14 +83,11 @@ def _save_outputs(adata, emb):
     d_model = int(emb.shape[1])
     index_arr = pa.array(adata.obs_names.astype(str).tolist(), type=pa.string())
     emb_arr = pa.array(emb.tolist(), type=pa.list_(pa.float32(), d_model))
-    table = pa.Table.from_arrays(
-        [index_arr, emb_arr],
-        names=[PARQUET_INDEX_COL, PARQUET_EMB_COL],
-        schema=pa.schema([
-            pa.field(PARQUET_INDEX_COL, pa.string()),
-            pa.field(PARQUET_EMB_COL, pa.list_(pa.float32(), d_model)),
-        ]),
-    )
+    schema = pa.schema([
+        pa.field(PARQUET_INDEX_COL, pa.string()),
+        pa.field(PARQUET_EMB_COL, pa.list_(pa.float32(), d_model)),
+    ])
+    table = pa.Table.from_arrays([index_arr, emb_arr], schema=schema)
     parquet_path = _unique_output("embs.parquet")
     pq.write_table(table, parquet_path, compression="zstd", use_dictionary=True)