From e91f163a82de2c5ed866ddff399ad6c439b2df55 Mon Sep 17 00:00:00 2001
From: light4ff <962977793@qq.com>
Date: Mon, 18 Mar 2024 16:58:26 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8Destimator=E6=A8=A1=E5=BC=8F?=
 =?UTF-8?q?=E4=B8=8B=E7=BB=AD=E8=AE=AD=E4=BF=9D=E5=AD=98=E7=AC=AC=E4=B8=80?=
 =?UTF-8?q?=E6=AD=A5=E6=A8=A1=E5=9E=8B=E6=97=B6=E6=89=BE=E4=B8=8D=E5=88=B0?=
 =?UTF-8?q?slice.data=E6=8A=A5=E9=94=99?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 mx_rec/saver/saver.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mx_rec/saver/saver.py b/mx_rec/saver/saver.py
index 196d744b..ffab7f7d 100644
--- a/mx_rec/saver/saver.py
+++ b/mx_rec/saver/saver.py
@@ -603,7 +603,7 @@ def merge_multi_files(upper_dir: str):
 
     Returns: None
     """
-    data_files = [file for file in os.listdir(upper_dir) if file.endswith(".data")]
+    data_files = [file for file in tf.io.gfile.listdir(upper_dir) if file.startswith("slice_")]
     data_files = sorted(data_files, key=os.path.basename)
     outfile_path = os.path.join(upper_dir, "slice.data")
     with tf.io.gfile.GFile(outfile_path, "wb") as outfile:
-- 
Gitee