Polars-python: is_in для lazyframe TypeError ⇐ Python

[Anonymous]

Я получаю следующую ошибку типа

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

Traceback (most recent call last):
File "/my/path/my_project/src/my_project/exploration/mre_lazyframe_error.py", line 39, in 
current.with_columns(pl.col("foo_bar").is_in(reference["foo_bar"]))
File "/my/path/.cache/pypoetry/virtualenvs/my_project-p95GORRi-py3.10/lib/python3.10/site-packages/polars/lazyframe/frame.py", line 619, in __getitem__
raise TypeError(msg)
TypeError: 'LazyFrame' object is not subscriptable (aside from slicing)

MRE

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

import numpy as np
import polars as pl

num_rows = 10000
ids = np.arange(num_rows)
foo_bar = np.random.randint(1, 101, num_rows)

current = pl.LazyFrame(
{
"id": ids,
"foo_bar": foo_bar,
}
)
reference = pl.LazyFrame(
{
"id": ids,
"foo_bar": np.random.randint(
1, 101, num_rows
),  # different random numbers for 'reference'
}
)

current.with_columns(
pl.col("foo_bar").is_in(reference["foo_bar"]).name.suffix("_avail"),
)

current.with_columns(pl.col("foo_bar").is_in(reference["foo_bar"]))

Насколько я понимаю, это должно быть возможно (is_in docs).
Когда я делаю это в режиме ожидания, оно выполняется без каких-либо ошибок.
Однако я бы сделал это желательно вычислять все отложенно.
Есть ли способ заставить это работать?


Подробнее здесь: https://stackoverflow.com/questions/790 ... -typeerror