From 2d27492e6e0a46b6e3ff1b9e4e24450b0e234927 Mon Sep 17 00:00:00 2001
From: abhinavmuk04 <69162586+abhinavmuk04@users.noreply.github.com>
Date: Thu, 20 Jun 2024 17:17:26 -0700
Subject: [PATCH] Treat 0 Estimations as NaN

---
 .../java/com/facebook/presto/cost/PlanNodeStatsEstimate.java    | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/presto-main/src/main/java/com/facebook/presto/cost/PlanNodeStatsEstimate.java b/presto-main/src/main/java/com/facebook/presto/cost/PlanNodeStatsEstimate.java
index ab6bf3c397bf7..0feb6b793efa4 100644
--- a/presto-main/src/main/java/com/facebook/presto/cost/PlanNodeStatsEstimate.java
+++ b/presto-main/src/main/java/com/facebook/presto/cost/PlanNodeStatsEstimate.java
@@ -102,7 +102,7 @@ public PlanNodeStatsEstimate(double outputRowCount, double totalSize, PMap<Varia
             JoinNodeStatsEstimate joinNodeStatsEstimate, TableWriterNodeStatsEstimate tableWriterNodeStatsEstimate, PartialAggregationStatsEstimate partialAggregationStatsEstimate)
     {
         checkArgument(isNaN(outputRowCount) || outputRowCount >= 0, "outputRowCount cannot be negative");
-        this.outputRowCount = outputRowCount;
+        this.outputRowCount = outputRowCount == 0 ? NaN : outputRowCount;
         this.totalSize = totalSize;
         this.variableStatistics = variableStatistics;
         this.sourceInfo = requireNonNull(sourceInfo, "SourceInfo is null");