Query Dataset

Execute Polars code on a dataset server-side.

The code runs in a sandboxed environment with access to:

• df: Polars DataFrame containing the dataset
• pl: Polars module for expressions

Assign the result to a result variable to return it. Multiline code is fully supported — use intermediate variables, comments, etc.

Security: Only allowlisted Polars DataFrame/Expression methods are permitted. Control-flow (for/while/if-stmt), comprehensions, f-strings, str.format, imports, eval/exec, and dunder access are all blocked.

Examples::

result = df.head(10)

filtered = df.filter(pl.col('score') > 0.5)
grouped = filtered.group_by('label').agg([
    pl.col('score').mean().alias('avg_score'),
    pl.col('score').count().alias('count')
])
result = grouped.sort('avg_score', descending=True)