Skip to content

Commit d58f447

Browse files
committed
Fix test_df_unpivot to work with Spark 4.0
1 parent bbd86af commit d58f447

File tree

1 file changed

+8
-6
lines changed

1 file changed

+8
-6
lines changed

crates/connect/src/dataframe.rs

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -2445,12 +2445,14 @@ mod tests {
24452445

24462446
let df = spark.create_dataframe(&data)?;
24472447

2448-
let df = df.unpivot(
2449-
[col("id")],
2450-
Some(vec![col("int"), col("float")]),
2451-
"var",
2452-
"val",
2453-
);
2448+
let df = df
2449+
.unpivot(
2450+
[col("id")],
2451+
Some(vec![col("int"), col("float")]),
2452+
"var",
2453+
"val",
2454+
)
2455+
.select(vec![col("id"), col("var"), col("val").cast("float")]);
24542456

24552457
let res = df.collect().await?;
24562458

0 commit comments

Comments
 (0)