update semdedup test

Signed-off-by: Sarah Yurick <[email protected]>
NVIDIA · Feb 25, 2025 · 349d586 · 349d586
1 parent 1801715
commit 349d586
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_semdedup.py b/tests/test_semdedup.py
@@ -126,9 +126,9 @@ def test_no_sem_dedup(
 
         sem_duplicates = SemDedup(
             config=config,
-            input_column="text",
-            id_column="doc_id",
-            id_column_type="str",
+            text_field="text",
+            id_field="doc_id",
+            id_field_type="str",
         )
 
         non_dedup_data_len = non_dedup_data.df.shape[0].compute()