Adding a special case to _common so that Data Ops get a more informative message

CHANGES.rst

@@ -16,6 +16,8 @@ Changes
   removed in the next release of skrub.
   :pr:`1546` by :user:`Vincent Maladiere <Vincent-Maladiere>`.
 
+- Improved error messages when a DataOp is being sent to dispatched functions.
+  :pr:`1607` by :user:`Riccardo Cappuzzo<rcap107>`.
 
 Bugfixes
 --------

skrub/_dataframe/_common.py

@@ -119,6 +119,14 @@
 
 
 def _raise(obj, kind="object"):
+    from .._data_ops._data_ops import DataOp
+
+    if isinstance(obj, DataOp):
+        raise TypeError(
+            "Operation not supported on DataOps. Expecting a Pandas or Polars "
+            f"{kind}, but got a DataOp. You may want to use `.skb.eval()` to "
+            "evaluate the DataOp and obtain a dataframe."
+        )
     raise TypeError(
         "Operation not supported on this object. Expecting a Pandas or Polars "
         f"{kind}, but got an object of type {type(obj)}."

skrub/_dataframe/tests/test_common.py

@@ -15,6 +15,7 @@
 from numpy.testing import assert_array_equal
 from pandas.testing import assert_frame_equal as pd_assert_frame_equal
 
+import skrub
 from skrub import selectors as s
 from skrub._dataframe import _common as ns
 
@@ -40,6 +41,12 @@ def test_not_implemented():
         params = [None] * n_params
         with pytest.raises(TypeError):
             func(*params)
+        dop = [skrub.var("a")] * n_params
+        with pytest.raises(
+            TypeError,
+            match=r"Operation not supported on DataOps.*Expecting a Pandas*",
+        ):
+            func(*dop)
 
 
 #