From 9173db391593a0abe7d5193f7014973357efec70 Mon Sep 17 00:00:00 2001 From: Sarah Yurick <53962159+sarahyurick@users.noreply.github.com> Date: Fri, 22 Nov 2024 10:54:00 -0800 Subject: [PATCH] fix typo (#388) Signed-off-by: Sarah Yurick --- nemo_curator/scripts/semdedup/extract_dedup_data.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/nemo_curator/scripts/semdedup/extract_dedup_data.py b/nemo_curator/scripts/semdedup/extract_dedup_data.py index c3dc0b0a5..8da1be763 100755 --- 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