Skip to content

Commit 5f3ecd6

Browse files
Fix CategoricalDtype docstring (#14622)
This PR fixes the docstring in CategoricalDtype where the repr has added a new field (categories_dtype). This PR fixes 2 doctest failures.
1 parent 0e83e20 commit 5f3ecd6

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

python/cudf/cudf/core/dtypes.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -228,11 +228,11 @@ def from_pandas(cls, dtype: pd.CategoricalDtype) -> "CategoricalDtype":
228228
>>> import pandas as pd
229229
>>> pd_dtype = pd.CategoricalDtype(categories=['b', 'a'], ordered=True)
230230
>>> pd_dtype
231-
CategoricalDtype(categories=['b', 'a'], ordered=True)
231+
CategoricalDtype(categories=['b', 'a'], ordered=True, categories_dtype=object)
232232
>>> cudf_dtype = cudf.CategoricalDtype.from_pandas(pd_dtype)
233233
>>> cudf_dtype
234-
CategoricalDtype(categories=['b', 'a'], ordered=True)
235-
"""
234+
CategoricalDtype(categories=['b', 'a'], ordered=True, categories_dtype=object)
235+
""" # noqa: E501
236236
return CategoricalDtype(
237237
categories=dtype.categories, ordered=dtype.ordered
238238
)
@@ -246,10 +246,10 @@ def to_pandas(self) -> pd.CategoricalDtype:
246246
>>> import cudf
247247
>>> dtype = cudf.CategoricalDtype(categories=['b', 'a'], ordered=True)
248248
>>> dtype
249-
CategoricalDtype(categories=['b', 'a'], ordered=True)
249+
CategoricalDtype(categories=['b', 'a'], ordered=True, categories_dtype=object)
250250
>>> dtype.to_pandas()
251-
CategoricalDtype(categories=['b', 'a'], ordered=True)
252-
"""
251+
CategoricalDtype(categories=['b', 'a'], ordered=True, categories_dtype=object)
252+
""" # noqa: E501
253253
if self._categories is None:
254254
categories = None
255255
else:

0 commit comments

Comments
 (0)