From 4cdd6b24c3f618f3c84840cd13353b893378eff4 Mon Sep 17 00:00:00 2001 From: Theodore Vasiloudis Date: Wed, 8 Nov 2023 20:52:12 +0000 Subject: [PATCH] Review comment --- .../graphstorm_processing/repartition_files.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/graphstorm-processing/graphstorm_processing/repartition_files.py b/graphstorm-processing/graphstorm_processing/repartition_files.py index 47505972c4..30075de73b 100644 --- a/graphstorm-processing/graphstorm_processing/repartition_files.py +++ b/graphstorm-processing/graphstorm_processing/repartition_files.py @@ -144,7 +144,8 @@ def write_parquet_to_relative_path( def create_new_relative_path_from_existing( original_relative_path: str, repartitioned_file_index: int, suffix: str = None ) -> str: - """Changes the index in the `original_relative_path` to `original_relative_path`. + """Changes the index of the filename ``part-`` in `original_relative_path` + to the one given in `repartitioned_file_index`. Given a path of the form 'path/to/parquet/part-00001-filename.snappy.parquet', changes the numerical part of the filename to match the provided `repartitioned_file_index`,