From d9ad40bf6f5d344b2b07ba1a1b5a819bd4bb836d Mon Sep 17 00:00:00 2001
From: speech_asr <wangjiaming.wjm@alibaba-inc.com>
Date: 星期一, 17 四月 2023 11:45:41 +0800
Subject: [PATCH] update

---
 funasr/utils/prepare_data.py |    9 +++------
 1 files changed, 3 insertions(+), 6 deletions(-)

diff --git a/funasr/utils/prepare_data.py b/funasr/utils/prepare_data.py
index a8c96eb..a0d97f6 100644
--- a/funasr/utils/prepare_data.py
+++ b/funasr/utils/prepare_data.py
@@ -63,13 +63,10 @@
     if os.path.exists(shape_path):
         print('Shape file for small dataset already exists.')
         return
-    os.makedirs(shape_path, exist_ok=True)
+
     split_shape_path = os.path.join(args.data_dir, dataset, "shape_files")
-    if os.path.exists(shape_path):
-        assert os.path.exists(os.path.join(args.data_dir, dataset, "speech_shape"))
-        print('Shape file for small dataset already exists.')
-        return
-    os.makedirs(shape_path, exist_ok=True)
+    if os.path
+    os.makedirs(split_shape_path, exist_ok=True)
 
     # split
     wav_scp_file = os.path.join(args.data_dir, dataset, "wav.scp")

--
Gitblit v1.9.1