From ef794131ec4f76eb93f6c2f706d1592b8a58b9ee Mon Sep 17 00:00:00 2001
From: zhangbutao <zhangbutao@cmss.chinamobile.com>
Date: Wed, 29 May 2024 12:10:06 +0800
Subject: [PATCH] Refine

---
 .../org/apache/iceberg/mr/mapreduce/IcebergInputFormat.java  | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/iceberg/iceberg-handler/src/main/java/org/apache/iceberg/mr/mapreduce/IcebergInputFormat.java b/iceberg/iceberg-handler/src/main/java/org/apache/iceberg/mr/mapreduce/IcebergInputFormat.java
index 5a38e98ad11a..6e0a1398f0dc 100644
--- a/iceberg/iceberg-handler/src/main/java/org/apache/iceberg/mr/mapreduce/IcebergInputFormat.java
+++ b/iceberg/iceberg-handler/src/main/java/org/apache/iceberg/mr/mapreduce/IcebergInputFormat.java
@@ -230,10 +230,11 @@ private List<InputSplit> planInputSplits(Table table, Configuration conf, Execut
     long fromVersion = conf.getLong(InputFormatConfig.SNAPSHOT_ID_INTERVAL_FROM, -1);
     Scan<?, FileScanTask, CombinedScanTask> scan;
     if (fromVersion != -1) {
-      scan = applyConfig(conf, createIncrementalAppendScan(table, conf)).planWith(workerPool);
+      scan = applyConfig(conf, createIncrementalAppendScan(table, conf));
     } else {
-      scan = applyConfig(conf, createTableScan(table, conf)).planWith(workerPool);
+      scan = applyConfig(conf, createTableScan(table, conf));
     }
+    scan = (Scan<?, FileScanTask, CombinedScanTask>) scan.planWith(workerPool);
 
     boolean allowDataFilesWithinTableLocationOnly =
         conf.getBoolean(HiveConf.ConfVars.HIVE_ICEBERG_ALLOW_DATAFILES_IN_TABLE_LOCATION_ONLY.varname,