Don't push sample correction through limit

Author: jan-elastic

x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/optimizer/rules/logical/ApplySampleCorrections.java

@@ -11,6 +11,7 @@
 import org.elasticsearch.xpack.esql.core.util.Holder;
 import org.elasticsearch.xpack.esql.expression.function.aggregate.HasSampleCorrection;
 import org.elasticsearch.xpack.esql.plan.logical.Aggregate;
+import org.elasticsearch.xpack.esql.plan.logical.Limit;
 import org.elasticsearch.xpack.esql.plan.logical.LogicalPlan;
 import org.elasticsearch.xpack.esql.plan.logical.Sample;
 import org.elasticsearch.xpack.esql.rule.Rule;
@@ -24,6 +25,9 @@ public LogicalPlan apply(LogicalPlan logicalPlan) {
             if (plan instanceof Sample sample) {
                 sampleProbability.set(sample.probability());
             }
+            if (plan instanceof Limit) {
+                sampleProbability.set(null);
+            }
             if (plan instanceof Aggregate && sampleProbability.get() != null) {
                 plan = plan.transformExpressionsOnly(
                     e -> e instanceof HasSampleCorrection hsc && hsc.isSampleCorrected() == false