Skip to content

Commit d67e073

Browse files
authored
Merge pull request rapidsai#15799 from rapidsai/branch-24.06
Forward-merge branch-24.06 into branch-24.08
2 parents 6c2c296 + b4daa16 commit d67e073

File tree

2 files changed

+14
-1
lines changed

2 files changed

+14
-1
lines changed

python/cudf/cudf/core/column/categorical.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -1439,7 +1439,7 @@ def as_ordered(self, ordered: bool):
14391439
categories=self.categories,
14401440
codes=self.codes,
14411441
mask=self.base_mask,
1442-
size=self.base_size,
1442+
size=self.size,
14431443
offset=self.offset,
14441444
ordered=ordered,
14451445
)

python/cudf/cudf/tests/test_indexing.py

+13
Original file line numberDiff line numberDiff line change
@@ -2349,3 +2349,16 @@ def test_loc_datetime_random_with_ts(data, scalar):
23492349
expected = pdf.loc[:i]
23502350

23512351
assert_eq(actual, expected)
2352+
2353+
2354+
def test_sliced_categorical_as_ordered():
2355+
df = cudf.DataFrame({"a": list("caba"), "b": list(range(4))})
2356+
df["a"] = df["a"].astype("category")
2357+
df = df.iloc[:2]
2358+
result = df["a"].cat.as_ordered()
2359+
expected = cudf.Series(
2360+
["c", "a"],
2361+
dtype=cudf.CategoricalDtype(list("abc"), ordered=True),
2362+
name="a",
2363+
)
2364+
assert_eq(result, expected)

0 commit comments

Comments
 (0)