fix import _raise

Author: abenechehab

skrub/_agg_joiner.py

@@ -16,7 +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._dataframe._common import _raise_dispatch_unregistered_type as _sbd_raise
 from skrub._dispatch import dispatch
 
 from ._check_input import CheckInputDataFrame
@@ -91,7 +91,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")
+    _sbd_raise(table, kind="DataFrame")
 
 
 @perform_groupby.specialize("pandas", argument_type="DataFrame")

skrub/_clean_categories.py

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

skrub/_clean_null_strings.py

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

skrub/_join_utils.py

@@ -266,9 +266,9 @@ def _do_left_join(left, right, left_on, right_on):
     # This function is accessed only when the first argument is neither
     # a pandas or polars DataFrame.
     # To avoid circular import:
-    from skrub._dataframe._common import _raise as _sbd_raise
+    from skrub._dataframe._common import _raise_dispatch_unregistered_type as _sbd_raise
 
-    raise _sbd_raise(left, kind="DataFrame")
+    _sbd_raise(left, kind="DataFrame")
 
 
 @_do_left_join.specialize("pandas", argument_type="DataFrame")

skrub/_reporting/_sample_table.py

@@ -65,7 +65,7 @@
 import pandas as pd
 
 from .. import _dataframe as sbd
-from .._dataframe._common import _raise as _sbd_raise
+from .._dataframe._common import _raise_dispatch_unregistered_type as _sbd_raise
 from .._dispatch import dispatch
 
 __all__ = ["make_table"]
@@ -90,7 +90,7 @@ def make_table(df, max_top_slice_size=5, max_bottom_slice_size=5):
     A dictionary with all the information needed by the html template, see this
     module's docstring for details.
     """
-    raise _sbd_raise(df, kind="DataFrame")
+    _sbd_raise(df, kind="DataFrame")
 
 
 @make_table.specialize("pandas", argument_type="DataFrame")

skrub/_reporting/_utils.py

@@ -7,7 +7,7 @@
 import numpy as np
 
 from skrub import _dataframe as sbd
-from skrub._dataframe._common import _raise as _sbd_raise
+from skrub._dataframe._common import _raise_dispatch_unregistered_type as _sbd_raise
 from skrub._dispatch import dispatch
 
 
@@ -17,7 +17,7 @@ def get_dtype_name(column):
 
 @dispatch
 def to_dict(df):
-    raise _sbd_raise(df, kind="DataFrame")
+    _sbd_raise(df, kind="DataFrame")
 
 
 @to_dict.specialize("pandas", argument_type="DataFrame")

skrub/_to_datetime.py

@@ -8,7 +8,7 @@
 from . import _dataframe as sbd
 from . import selectors as s
 from ._apply_to_cols import RejectColumn, SingleColumnTransformer
-from ._dataframe._common import _raise as _sbd_raise
+from ._dataframe._common import _raise_dispatch_unregistered_type as _sbd_raise
 from ._dispatch import dispatch
 from ._wrap_transformer import wrap_transformer
 
@@ -19,7 +19,7 @@
 
 @dispatch
 def _get_time_zone(col):
-    raise _sbd_raise(col, kind="Series")
+    _sbd_raise(col, kind="Series")
 
 
 @_get_time_zone.specialize("pandas", argument_type="Column")
@@ -45,7 +45,7 @@ def _get_time_zone_polars(col):
 
 @dispatch
 def _convert_time_zone(col, time_zone):
-    raise _sbd_raise(col, kind="Series")
+    _sbd_raise(col, kind="Series")
 
 
 @_convert_time_zone.specialize("pandas", argument_type="Column")

skrub/selectors/_base.py

@@ -1,5 +1,5 @@
 from .. import _dataframe as sbd
-from .._dataframe._common import _raise as _sbd_raise
+from .._dataframe._common import _raise_dispatch_unregistered_type as _sbd_raise
 from .._dispatch import dispatch
 from .._utils import repr_args
 
@@ -160,7 +160,7 @@ def make_selector(obj):
 
 @dispatch
 def _select_col_names(df, col_names):
-    raise _sbd_raise(df, kind="DataFrame")
+    _sbd_raise(df, kind="DataFrame")
 
 
 @_select_col_names.specialize("pandas", argument_type="DataFrame")