Fix writer count error (#28)

Signed-off-by: sunby <[email protected]>
milvus-io · Jul 26, 2023 · 10dc26d · 10dc26d
1 parent 6df90c3
commit 10dc26d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cpp/src/format/parquet/file_writer.cpp b/cpp/src/format/parquet/file_writer.cpp
@@ -19,7 +19,7 @@ Status ParquetFileWriter::Init() {
 
 Status ParquetFileWriter::Write(arrow::RecordBatch* record) {
   RETURN_ARROW_NOT_OK(writer_->WriteRecordBatch(*record));
-  count_ += record->num_columns();
+  count_ += record->num_rows();
   return Status::OK();
 }