diff --git a/graphrag/storage/sql_server_pipeline_storage.py b/graphrag/storage/sql_server_pipeline_storage.py index 1ab12bf3..5426ab95 100644 --- a/graphrag/storage/sql_server_pipeline_storage.py +++ b/graphrag/storage/sql_server_pipeline_storage.py @@ -219,7 +219,7 @@ class SQLServerPipelineStorage(PipelineStorage): data.append(processed_row) # Create DataFrame from processed data - data_frame = pd.DataFrame(data, columns=columns) + data_frame = pd.DataFrame(data, columns=np.array(columns)) buffer = BytesIO() data_frame.to_parquet(buffer) buffer.seek(0)