sample() inside agg() seems to cause incorrect Schema for LazyFrame

[cmdlineluser]

Checks

• I have checked that this issue has not already been reported.
• I have confirmed this bug exists on the latest version of Polars.

Reproducible example

import polars as pl

df = pl.LazyFrame({"id": [1, 2], "val": [3, 4]})

(df.group_by("id")
   .agg(pl.col.val.sample(pl.len()))
   .with_columns(
       same = pl.col.val.list.set_intersection(pl.col.val)
   )
   .collect()
)
# InvalidOperationError: expected List type, got: i64

Log output

No response

Issue description

Should this not be List(Int64)?

(df.group_by("id")
   .agg(pl.col.val.sample(pl.len()))
   .collect_schema()["val"]
)
# Int64

An early collect shows the working query.

(df.group_by("id")
   .agg(pl.col.val.sample(pl.len()))
   .collect()
   .with_columns(
       same = pl.col.val.list.set_intersection(pl.col.val)
   )
)

# shape: (2, 3)
# ┌─────┬───────────┬───────────┐
# │ id  ┆ val       ┆ same      │
# │ --- ┆ ---       ┆ ---       │
# │ i64 ┆ list[i64] ┆ list[i64] │
# ╞═════╪═══════════╪═══════════╡
# │ 1   ┆ [3]       ┆ [3]       │
# │ 2   ┆ [4]       ┆ [4]       │
# └─────┴───────────┴───────────┘

But other .list functions seem to work OK, so not sure if it something specific to .list.set_* ones.

(df.group_by("id")
   .agg(pl.col.val.sample(pl.len()))
   .with_columns(
       len = pl.col.val.list.len()
   )
   .collect()
)

# shape: (2, 3)
# ┌─────┬───────────┬─────┐
# │ id  ┆ val       ┆ len │
# │ --- ┆ ---       ┆ --- │
# │ i64 ┆ list[i64] ┆ u32 │
# ╞═════╪═══════════╪═════╡
# │ 1   ┆ [3]       ┆ 1   │
# │ 2   ┆ [4]       ┆ 1   │
# └─────┴───────────┴─────┘

Expected behavior

No exception.

Installed versions

--------Version info---------
Polars:               1.6.0
Index type:           UInt32
Platform:             macOS-13.6.1-arm64-arm-64bit
Python:               3.12.2 (main, Feb  6 2024, 20:19:44) [Clang 15.0.0 (clang-1500.1.0.2.5)]

----Optional dependencies----
adbc_driver_manager:  <not installed>
cloudpickle:          <not installed>
connectorx:           <not installed>
deltalake:            <not installed>
fastexcel:            <not installed>
fsspec:               <not installed>
gevent:               <not installed>
great_tables:         <not installed>
hvplot:               <not installed>
matplotlib:           <not installed>
nest_asyncio:         <not installed>
numpy:                1.26.4
openpyxl:             <not installed>
pandas:               2.2.1
pyarrow:              15.0.2
pydantic:             <not installed>
pyiceberg:            <not installed>
sqlalchemy:           <not installed>
torch:                <not installed>
xlsx2csv:             <not installed>
xlsxwriter:           <not installed>

[cmdlineluser]

Fixed by #19016