change _raise to _raise_dispatch_unregistered_type

skrub/_agg_joiner.py

@@ -16,8 +16,7 @@
 from skrub import _dataframe as sbd
 from skrub import _join_utils, _utils
 from skrub import selectors as s
-from skrub._dataframe._common import _raise as _sbd_raise
-from skrub._dispatch import dispatch
+from skrub._dispatch import dispatch, raise_dispatch_unregistered_type
 
 from ._check_input import CheckInputDataFrame
 
@@ -91,7 +90,7 @@ def aggregate(table, key, cols_to_agg, operations, suffix):
 
 @dispatch
 def perform_groupby(table, key, cols_to_agg, operations):
-    raise _sbd_raise(table, kind="DataFrame")
+    raise_dispatch_unregistered_type(table, kind="DataFrame")
 
 
 @perform_groupby.specialize("pandas", argument_type="DataFrame")

skrub/_clean_categories.py

@@ -2,15 +2,14 @@
 
 from . import _dataframe as sbd
 from ._apply_to_cols import RejectColumn, SingleColumnTransformer
-from ._dataframe._common import _raise as _sbd_raise
-from ._dispatch import dispatch
+from ._dispatch import dispatch, raise_dispatch_unregistered_type
 
 __all__ = ["CleanCategories"]
 
 
 @dispatch
 def _with_string_categories(col):
-    raise _sbd_raise(col, kind="Series")
+    raise_dispatch_unregistered_type(col, kind="Series")
 
 
 @_with_string_categories.specialize("pandas", argument_type="Column")

skrub/_clean_null_strings.py

@@ -1,7 +1,6 @@
 from . import _dataframe as sbd
 from ._apply_to_cols import RejectColumn, SingleColumnTransformer
-from ._dataframe._common import _raise as _sbd_raise
-from ._dispatch import dispatch
+from ._dispatch import dispatch, raise_dispatch_unregistered_type
 
 __all__ = ["CleanNullStrings"]
 
@@ -32,7 +31,7 @@
 
 @dispatch
 def _trim_whitespace_only(col):
-    raise _sbd_raise(col, kind="Series")
+    raise_dispatch_unregistered_type(col, kind="Series")
 
 
 @_trim_whitespace_only.specialize("pandas", argument_type="Column")