fix(rust, python): fix pipeline dtypes (#5224)

pola-rs · Oct 16, 2022 · 3f1faeb · 3f1faeb
1 parent bac952e
commit 3f1faeb
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 30 deletions.
diff --git a/examples/read_csv/benches/streaming.rs b/examples/read_csv/benches/streaming.rs
diff --git a/polars/polars-lazy/polars-pipe/src/executors/sinks/groupby/aggregates/convert.rs b/polars/polars-lazy/polars-pipe/src/executors/sinks/groupby/aggregates/convert.rs
@@ -111,10 +111,10 @@ where
                     DataType::UInt8 => AggregateFunction::SumI64(SumAgg::<i64>::new()),
                     DataType::UInt16 => AggregateFunction::SumI64(SumAgg::<i64>::new()),
                     //  these stay true to there types
-                    DataType::UInt32 => AggregateFunction::SumI32(SumAgg::<i32>::new()),
-                    DataType::UInt64 => AggregateFunction::SumI64(SumAgg::<i64>::new()),
-                    DataType::Int32 => AggregateFunction::SumU32(SumAgg::<u32>::new()),
-                    DataType::Int64 => AggregateFunction::SumU64(SumAgg::<u64>::new()),
+                    DataType::UInt32 => AggregateFunction::SumU32(SumAgg::<u32>::new()),
+                    DataType::UInt64 => AggregateFunction::SumU64(SumAgg::<u64>::new()),
+                    DataType::Int32 => AggregateFunction::SumI32(SumAgg::<i32>::new()),
+                    DataType::Int64 => AggregateFunction::SumI64(SumAgg::<i64>::new()),
                     DataType::Float32 => AggregateFunction::SumF32(SumAgg::<f32>::new()),
                     DataType::Float64 => AggregateFunction::SumF64(SumAgg::<f64>::new()),
                     _ => unreachable!(),