Skip to content

Commit 5399904

Browse files
authored
Fix starting index for batch processing (#80)
1 parent 0ed25b3 commit 5399904

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/data_designer/engine/dataset_builders/column_wise_builder.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -88,8 +88,8 @@ def build(
8888
start_time = time.perf_counter()
8989

9090
self.batch_manager.start(num_records=num_records, buffer_size=buffer_size)
91-
for batch_idx in range(1, self.batch_manager.num_batches + 1):
92-
logger.info(f"⏳ Processing batch {batch_idx} of {self.batch_manager.num_batches}")
91+
for batch_idx in range(self.batch_manager.num_batches):
92+
logger.info(f"⏳ Processing batch {batch_idx + 1} of {self.batch_manager.num_batches}")
9393
self._run_batch(generators)
9494
df_batch = self._run_processors(
9595
stage=BuildStage.POST_BATCH,

0 commit comments

Comments
 (0)