check e.is_scalar

pola-rs · Jan 8, 2025 · de9b99a · de9b99a
1 parent 94b79ce
commit de9b99a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/crates/polars-expr/src/expressions/sortby.rs b/crates/polars-expr/src/expressions/sortby.rs
@@ -239,7 +239,7 @@ impl PhysicalExpr for SortByExpr {
 
                         if broadcast_length != column.len() {
                             polars_ensure!(
-                                broadcast_length == 1,
+                                broadcast_length == 1 && e.is_scalar(),
                                 expr = self.expr, ShapeMismatch:
                                 "`sort_by` produced different length ({}) than earlier Series' length in `by` ({})",
                                 broadcast_length, column.len()