remove typehint

Signed-off-by: Praateek <[email protected]>
NVIDIA · Oct 30, 2024 · 06f6085 · 06f6085
1 parent 7253f38
commit 06f6085
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nemo_curator/utils/fuzzy_dedup_utils/shuffle_utils.py b/nemo_curator/utils/fuzzy_dedup_utils/shuffle_utils.py
@@ -154,7 +154,7 @@ def get_shuffle_partition_info(
 
 
 def text_bytes_aware_shuffle(
-    df: dask_cuda.DataFrame,
+    df,
     partition_on: str,
     text_column: str,
     num_workers: Optional[int] = None,