From 1ade0ec2306afde0d804643e6d4c0950b795f9dc Mon Sep 17 00:00:00 2001
From: Mehdi Ali <33023925+mali-git@users.noreply.github.com>
Date: Tue, 21 Jan 2025 10:35:30 +0100
Subject: [PATCH] Update src/modalities/__main__.py
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Co-authored-by: Max Lübbering <2804731+le1nux@users.noreply.github.com>
---
 src/modalities/__main__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/modalities/__main__.py b/src/modalities/__main__.py
index 1141b0fa..82bdcede 100644
--- a/src/modalities/__main__.py
+++ b/src/modalities/__main__.py
@@ -192,7 +192,7 @@ def CMD_entry_point_merge_packed_data(src_paths: list[Path], target_path: Path):
 @click.option(
     "--batch-size", type=int, default=100, show_default=True, help="Number of documents to process per batch."
 )
-def shuffle_tokenized_data_entrypoint(input_data_path: Path, batch_size: int) -> None:
+def CMD_shuffle_tokenized_data(input_data_path: Path, batch_size: int) -> None:
     """Entrypoint for shuffling tokenized data.
 
     Args: