change format

dmlc · Oct 16, 2024 · 4573ea2 · 4573ea2
1 parent d77b911
commit 4573ea2
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/tools/distpartitioning/dataset_utils.py b/tools/distpartitioning/dataset_utils.py
@@ -560,9 +560,7 @@ def get_dataset(
                         if next_chunk is None:
                             break
 
-                        next_table = pyarrow.Table.from_batches(
-                            [next_chunk]
-                        )
+                        next_table = pyarrow.Table.from_batches([next_chunk])
                         src_ids.append(next_table["f0"].to_numpy())
                         dst_ids.append(next_table["f1"].to_numpy())
             elif (