Polars categorical feature and lazy api doesn't work like expected ⇐ Python

[Anonymous]

Я пытаюсь соединить два Dataframe с помощью категориальных функций и ленивого API. Я попробовал сделать так, как описано в руководстве пользователя (https://pola-rs.github.io/polars-book/user-guide/ Performance/strings.html)

Код: Выделить всё

count = admin_df.groupby(['admin','EVENT_DATE']).pivot(pivot_column='FIVE_TYPE',values_column='count').first().lazy()
fatalities = admin_df.groupby(['admin','EVENT_DATE']).pivot(pivot_column='FIVE_TYPE',values_column='FATALITIES').first().lazy()
fatalities = fatalities.with_column(pl.col("admin").cast(pl.Categorical))
count = count.with_column(pl.col("admin").cast(pl.Categorical))
admin_df = fatalities.join(count,on=['admin','EVENT_DATE']).collect()

но я получаю следующую ошибку:

Код: Выделить всё

    Traceback (most recent call last):
File "country_level.py", line 33, in 
country_level('/c/Users/Sebastian/feast/fluent_sunfish/data/ACLED_geocoded.parquet')
File "country_level.py", line 10, in country_level
country_df=aggregate_by_date(df)
File "country_level.py", line 29, in aggregate_by_date
admin_df = fatalities.join(count,on=['admin','EVENT_DATE']).collect()
File "/home/sebastian/.local/lib/python3.8/site-packages/polars/internals/lazy_frame.py", line 293, in collect
return pli.wrap_df(ldf.collect())
RuntimeError: Any(ValueError("joins on categorical dtypes can only happen if they are created under the same global string cache"))

с использованием pl.StringCache(): все работает нормально, хотя в руководстве пользователя сказано, что это не нужно, если вы используете ленивый API, я пропустил что-то или это ошибка?

Подробнее здесь: https://stackoverflow.com/questions/702 ... e-expected