Modin v0.32.0分组后无法访问category列的.cat.codes问题

问题现象 #

在Modin v0.32.0环境下，DataFrame的category类型列在执行groupby操作前能正常通过.cat.codes访问编码值，但分组后得到的子DataFrame中，该列无法再访问.cat.codes，执行时会抛出错误。

复现代码 #

import modin.pandas as pd

df = pd.read_parquet(path="../../sample.parquet")
print(df['DummyCol'].cat.codes)

data_segments = {}
for key, group in df.groupby("DATASEGMENT", sort=False):
    data_segments[key] = group

# 该行执行会抛出错误
print(data_segments["202201"]["DummyCol"].cat.codes)

相关说明 #

• 已准备可复现问题的数据集
• 已在Modin官方仓库提交Issue（编号：7586）

内容的提问来源于stack exchange，提问作者Sumukha G C