Fixed a few more redundancies.

rapidsai · Apr 27, 2021 · 51e6bea · 51e6bea
1 parent 0329ad5
commit 51e6bea
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 17 deletions.
diff --git a/python/cudf/cudf/core/column/column.py b/python/cudf/cudf/core/column/column.py
@@ -323,16 +323,14 @@ def to_arrow(self) -> pa.Array:
           4
         ]
         """
-        result = libcudf.interop.to_arrow(
+        return libcudf.interop.to_arrow(
             libcudf.table.Table(
                 cudf.core.column_accessor.ColumnAccessor({"None": self})
             ),
             [["None"]],
             keep_index=False,
         )["None"].chunk(0)
 
-        return result
-
     @classmethod
     def from_arrow(cls, array: pa.Array) -> ColumnBase:
         """

diff --git a/python/cudf/cudf/core/column/datetime.py b/python/cudf/cudf/core/column/datetime.py
@@ -139,15 +139,12 @@ def to_pandas(
         # https://issues.apache.org/jira/browse/ARROW-9772
 
         # Pandas supports only `datetime64[ns]`, hence the cast.
-        pd_series = pd.Series(
-            self.astype("datetime64[ns]").to_array("NAT"), copy=False
+        return pd.Series(
+            self.astype("datetime64[ns]").to_array("NAT"),
+            copy=False,
+            index=index,
         )
 
-        if index is not None:
-            pd_series.index = index
-
-        return pd_series
-
     def get_dt_field(self, field: str) -> ColumnBase:
         return libcudf.datetime.extract_datetime_component(self, field)
 

diff --git a/python/cudf/cudf/core/column/string.py b/python/cudf/cudf/core/column/string.py
@@ -4788,13 +4788,7 @@ def to_arrow(self) -> pa.Array:
                 pa.null(), len(self), [pa.py_buffer((b""))]
             )
         else:
-            return libcudf.interop.to_arrow(
-                libcudf.table.Table(
-                    cudf.core.column_accessor.ColumnAccessor({"None": self})
-                ),
-                [["None"]],
-                keep_index=False,
-            )["None"].chunk(0)
+            return super().to_arrow()
 
     def sum(
         self, skipna: bool = None, dtype: Dtype = None, min_count: int = 0