Implement rich_repr for expressions

[TomAugspurger]

Description

This adds a __rich_repr__ implmementation for cudf-polars' internal data structures Expr and PartitionInfo that should be a bit easier to read while debugging. There's no dependency (required or optional) on rich. There's no change to the regular repr used by Python to print objects. To see the change, use rich.pretty.print.

Here's a screenshot of the output:

import polars as pl
import rich.pretty
from cudf_polars.dsl.translate import Translator

ldf = pl.LazyFrame({"a": ["a", "b", "c", "d", "a", None, "b", "c"]})
q = ldf.group_by((pl.col("a").str.starts_with("a"))).agg(
    pl.col("a").count().alias("count")
)

ir = Translator(q._ldf.visit(), engine=pl.GPUEngine()).translate_ir()

print("default")
print("-" * 80)
print(ir, end="\n\n")

print("rich")
print("-" * 80)
rich.pretty.pprint(ir)

[copy-pr-bot]

Auto-sync is disabled for draft pull requests in this repository. Workflows must be run manually.

Contributors can view more details about this message here.

[TomAugspurger]

/merge