Fix validation of queries with Sample

polarsignals · May 30, 2024 · bd300cd · bd300cd
1 parent 05c806b
commit bd300cd
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/query/logicalplan/logicalplan.go b/query/logicalplan/logicalplan.go
@@ -160,6 +160,13 @@ func (plan *LogicalPlan) DataTypeForExpr(expr Expr) (arrow.DataType, error) {
 			return nil, fmt.Errorf("data type for expr %v within Distinct: %w", expr, err)
 		}
 
+		return t, nil
+	case plan.Sample != nil:
+		t, err := expr.DataType(plan.Input)
+		if err != nil {
+			return nil, fmt.Errorf("data type for expr %v within Sample: %w", expr, err)
+		}
+
 		return t, nil
 	default:
 		return nil, fmt.Errorf("unknown logical plan")