change dataset_utils

dmlc · Oct 17, 2024 · b47898f · b47898f
1 parent 72ecb68
commit b47898f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tools/distpartitioning/dataset_utils.py b/tools/distpartitioning/dataset_utils.py
@@ -548,9 +548,9 @@ def get_dataset(
                 )
                 parse_options = pyarrow.csv.ParseOptions(delimiter=" ")
 
-                # if getsize() == 0, the file is empty, indicating that the partition doesn't have this attribute.
-                # The src_ids and dst_ids should remain empty.
                 if os.path.getsize(edge_file) == 0:
+                    # if getsize() == 0, the file is empty, indicating that the partition doesn't have this attribute.
+                    # The src_ids and dst_ids should remain empty.
                     continue
                 with pyarrow.csv.open_csv(
                     edge_file,