chore: improve docstring

apache · Oct 13, 2024 · b39a5f0 · b39a5f0
1 parent bded1c4
commit b39a5f0
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/python/datafusion/dataframe.py b/python/datafusion/dataframe.py
@@ -246,11 +246,10 @@ def sort(self, *exprs: Expr | SortExpr) -> DataFrame:
         return DataFrame(self.df.sort(*exprs_raw))
 
     def cast(self, mapping: dict[str, pa.DataType[Any]]) -> DataFrame:
-        """Cast all or a subset of columns to new dtype.
+        """Cast one or more columns to a different data type.
 
         Args:
-            mapping (dict[str, pa.DataType[Any]]):  Mapped with column as key and column
-                dtype as value.
+            mapping: Mapped with column as key and column dtype as value.
 
         Returns:
             DataFrame after casting columns