Fix last copy_construct issue.

rapidsai · Aug 5, 2021 · 6727fc4 · 6727fc4
1 parent 71db785
commit 6727fc4
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/python/cudf/cudf/core/indexing.py b/python/cudf/cudf/core/indexing.py
@@ -98,10 +98,9 @@ def __getitem__(self, arg):
             or _is_null_host_scalar(data)
         ):
             return data
-        # return self._sr._from_data(
-        #     {self._sr.name: data}, self._sr.index.take(arg))
-        index = self._sr.index.take(arg)
-        return self._sr._copy_construct(data=data, index=index)
+        return self._sr._from_data(
+            {self._sr.name: data}, index=cudf.Index(self._sr.index.take(arg))
+        )
 
     def __setitem__(self, key, value):
         from cudf.core.column import column