update doctests

skrub-data · Nov 22, 2023 · 4b2e020 · 4b2e020
1 parent bbd88f9
commit 4b2e020
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/skrub/_fuzzy_join.py b/skrub/_fuzzy_join.py
@@ -133,6 +133,7 @@ def fuzzy_join(
     ...     on="Country",
     ...     suffix="_capitals",
     ...     max_dist=1.0,
+    ...     insert_match_info=False,
     ... )
       Country Country_capitals Capital_capitals
     0  France           France            Paris
@@ -145,6 +146,7 @@ def fuzzy_join(
     ...     suffix="_capitals",
     ...     drop_unmatched=True,
     ...     max_dist=1.0,
+    ...     insert_match_info=False,
     ... )
       Country Country_capitals Capital_capitals
     0  France           France            Paris
@@ -155,6 +157,7 @@ def fuzzy_join(
     ...     on="Country",
     ...     suffix="_capitals",
     ...     max_dist=float("inf"),
+    ...     insert_match_info=False,
     ... )
       Country Country_capitals Capital_capitals
     0  France           France            Paris

diff --git a/skrub/_joiner.py b/skrub/_joiner.py
@@ -176,7 +176,13 @@ class Joiner(TransformerMixin, BaseEstimator):
     0  Germany  Berlin
     1   France   Paris
     2    Italy    Rome
-    >>> joiner = Joiner(aux_table, key="Country", suffix="_capitals", max_dist=0.9)
+    >>> joiner = Joiner(
+    ...     aux_table,
+    ...     key="Country",
+    ...     suffix="_capitals",
+    ...     max_dist=0.9,
+    ...     insert_match_info=False,
+    ... )
     >>> joiner.fit_transform(main_table)
       Country Country_capitals Capital_capitals
     0  France           France            Paris