From 4e461adfc329a40af47aaabca3d74db91684f661 Mon Sep 17 00:00:00 2001
From: Ning Sun <sunning@greptime.com>
Date: Fri, 22 May 2026 00:52:07 +0800
Subject: [PATCH] fix: revert changes on flatsource

---
 src/mito2/src/read.rs | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/mito2/src/read.rs b/src/mito2/src/read.rs
index b671152dee..567754f939 100644
--- a/src/mito2/src/read.rs
+++ b/src/mito2/src/read.rs
@@ -1163,7 +1163,7 @@ impl FlatSource {
         }
     }
 
-    pub fn schema(&self) -> &SchemaRef {
+    pub(crate) fn schema(&self) -> &SchemaRef {
         &self.schema
     }
 
@@ -1171,6 +1171,7 @@ impl FlatSource {
         self.inner.next_batch().await
     }
 
+    #[cfg(test)]
     pub(crate) fn take_iter(self) -> BoxedRecordBatchIterator {
         match self.inner {
             FlatSourceInner::Iter(iter) => iter,