fix test

pola-rs · Oct 11, 2024 · 86305a1 · 86305a1
1 parent 3239f9c
commit 86305a1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/crates/polars/tests/it/lazy/queries.rs b/crates/polars/tests/it/lazy/queries.rs
@@ -203,7 +203,7 @@ fn test_apply_multiple_columns() -> PolarsResult<()> {
         .select([map_multiple(
             multiply,
             [col("A"), col("B")],
-            GetOutput::from_type(DataType::Float64),
+            GetOutput::from_type(DataType::Int32),
         )])
         .collect()?;
     let out = out.column("A")?;
@@ -219,7 +219,7 @@ fn test_apply_multiple_columns() -> PolarsResult<()> {
         .agg([apply_multiple(
             multiply,
             [col("A"), col("B")],
-            GetOutput::from_type(DataType::Float64),
+            GetOutput::from_type(DataType::Int32),
             true,
         )])
         .collect()?;