Как гарантировать, что два модуля используют один и тот же строковый кеш? ⇐ Python

[Anonymous]

Мне хотелось бы, чтобы фреймы данных Polars в нескольких модулях использовали один и тот же строковый кеш, но мне сложно с этим справиться. Представьте, что у меня есть пакет с модулями A и B:

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

# A.py
import polars as pl

df_A = pl.DataFrame({
'A': pl.Series(['a', 'b', 'c'], dtype=pl.Category)
})

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

# B.py
import polars as pl

df_B = pl.DataFrame({
'A': pl.Series(['a', 'b', 'c'], dtype=pl.Category),
'B': pl.Series(['d', 'e', 'f'], dtype=pl.Category)
})

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

# __init__.py

from . import A, B

pl.toggle_string_cache(True)

df_C = A.df_A.join(B.df_B, on='A')

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

exceptions.ComputeError: joins/or comparisons on categoricals can only happen if they were created under the same global string cache

Это сработает, если я перенесу импорт после вызова pl.toggle_string_cache(True), но это плохой дизайн кода: импорт всегда должен идти первым. Я могу подумать о перемещении A.df_A и B.df_B в глобальные переменные с помощью функций-генераторов, но я надеялся на более элегантное решение.


Подробнее здесь: https://stackoverflow.com/questions/759 ... ring-cache