Make alias for at/iat

rapidsai · Jul 5, 2024 · 97a283c · 97a283c
1 parent 8293be7
commit 97a283c
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/python/cudf/cudf/core/dataframe.py b/python/cudf/cudf/core/dataframe.py
@@ -462,6 +462,9 @@ def _setitem_tuple_arg(self, key, value):
                             self._frame[col].loc[key[0]] = value[i]
 
 
+_DataFrameAtIndexer = _DataFrameLocIndexer
+
+
 class _DataFrameIlocIndexer(_DataFrameIndexer):
     """
     For selection by index.
@@ -584,6 +587,9 @@ def _setitem_tuple_arg(self, key, value):
                         self._frame[col].iloc[key[0]] = value[i]
 
 
+_DataFrameiAtIndexer = _DataFrameIlocIndexer
+
+
 class DataFrame(IndexedFrame, Serializable, GetAttrGetItemMixin):
     """
     A GPU Dataframe object.
@@ -2581,14 +2587,14 @@ def iat(self):
         """
         Alias for ``DataFrame.iloc``; provided for compatibility with Pandas.
         """
-        return self.iloc
+        return _DataFrameiAtIndexer(self)
 
     @property
     def at(self):
         """
         Alias for ``DataFrame.loc``; provided for compatibility with Pandas.
         """
-        return self.loc
+        return _DataFrameAtIndexer(self)
 
     @property  # type: ignore
     @_external_only_api(

diff --git a/python/cudf/cudf/pandas/_wrappers/pandas.py b/python/cudf/cudf/pandas/_wrappers/pandas.py
@@ -777,13 +777,13 @@ def Index__new__(cls, *args, **kwargs):
 
 _AtIndexer = make_intermediate_proxy_type(
     "_AtIndexer",
-    _Unusable(),
+    cudf.core.dataframe._DataFrameAtIndexer,
     pd.core.indexing._AtIndexer,
 )
 
 _iAtIndexer = make_intermediate_proxy_type(
     "_iAtIndexer",
-    _Unusable(),
+    cudf.core.dataframe._DataFrameiAtIndexer,
     pd.core.indexing._iAtIndexer,
 )