fix typo (NVIDIA#388)

Signed-off-by: Sarah Yurick <sarahyurick@gmail.com> Signed-off-by: Vinay Raman <viraman@nvidia.com>
vinay-raman · Nov 26, 2024 · 756072b · 756072b
1 parent cb49b1f
commit 756072b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nemo_curator/scripts/semdedup/extract_dedup_data.py b/nemo_curator/scripts/semdedup/extract_dedup_data.py
@@ -36,8 +36,8 @@ def main(args):
         sorted_clusters_dir=os.path.join(
             cache_dir, semdedup_config.clustering_save_loc, "sorted"
         ),
-        id_col=args.id_column,
-        id_col_type=args.id_column_type,
+        id_column=args.id_column,
+        id_column_type=args.id_column_type,
         which_to_keep=semdedup_config.which_to_keep,
         output_dir=os.path.join(
             semdedup_config.cache_dir, semdedup_config.clustering_save_loc