lint

ray-project · Oct 22, 2024 · 54660f3 · 54660f3
1 parent e80f2e0
commit 54660f3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/ray/data/dataset.py b/python/ray/data/dataset.py
@@ -713,6 +713,7 @@ def add_column(
 
             >>> import ray
             >>> import pyarrow as pa
+            >>> import pyarrow.compute as pc
             >>> ds = ray.data.range(100)
             >>> ds.schema()
             Column  Type
@@ -721,7 +722,7 @@ def add_column(
 
             Add a new column equal to ``id * 2``.
 
-            >>> ds.add_column("new_id", lambda x: pa.compute.multiply(x["id"], 2)).schema()
+            >>> ds.add_column("new_id", lambda x: pc.multiply(x["id"], 2)).schema()
             Column  Type
             ------  ----
             id      int64