Polars has a lazy API, which allows for query optimization. Currently, AgentSetPolars and PolarsMixins are implemented with the eager API, which may lead to a significant performance loss. Collection could be done at each step.

References:

• https://docs.pola.rs/user-guide/concepts/lazy-vs-eager/
• https://stackoverflow.com/questions/76612163/what-are-the-advantages-of-a-polars-lazyframe-over-a-dataframe