From 80e6c258cf89b5f11f4e52a4cc5a9cf2e95aa7be Mon Sep 17 00:00:00 2001
From: Yuekai Zhang <zhangyuekai@foxmail.com>
Date: 星期一, 06 三月 2023 16:48:02 +0800
Subject: [PATCH] update token list

---
 egs/alimeeting/diarization/sond/unit_test.py |    6 +++---
 1 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/egs/alimeeting/diarization/sond/unit_test.py b/egs/alimeeting/diarization/sond/unit_test.py
index 10fc558..84a4247 100644
--- a/egs/alimeeting/diarization/sond/unit_test.py
+++ b/egs/alimeeting/diarization/sond/unit_test.py
@@ -48,7 +48,7 @@
     diar_model_path = "sond.pth"
     output_dir = "./outputs"
     data_path_and_name_and_type = [
-        ("data/unit_test/test_wav.scp", "speech", "kaldi_ark"),
+        ("data/unit_test/test_wav.scp", "speech", "sound"),
         ("data/unit_test/test_profile.scp", "profile", "kaldi_ark"),
     ]
     pipeline = inference_launch(
@@ -68,13 +68,13 @@
     diar_config_path = "config.yaml"
     diar_model_path = "sond.pth"
     output_dir = "./outputs"
-    raw_inputs = [
+    raw_inputs = [[
         "data/unit_test/raw_inputs/record.wav",
         "data/unit_test/raw_inputs/spk1.wav",
         "data/unit_test/raw_inputs/spk2.wav",
         "data/unit_test/raw_inputs/spk3.wav",
         "data/unit_test/raw_inputs/spk4.wav"
-    ]
+    ]]
     pipeline = inference_launch(
         mode="sond_demo",
         diar_train_config=diar_config_path,

--
Gitblit v1.9.1