From a1b0cd33d50cee3e4612d1e787399e508b453a4a Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期四, 21 十二月 2023 14:20:21 +0800
Subject: [PATCH] rename register tables
---
funasr/datasets/audio_datasets/datasets.py | 10 +++++-----
1 files changed, 5 insertions(+), 5 deletions(-)
diff --git a/funasr/datasets/audio_datasets/datasets.py b/funasr/datasets/audio_datasets/datasets.py
index d69d0b5..bfdf86a 100644
--- a/funasr/datasets/audio_datasets/datasets.py
+++ b/funasr/datasets/audio_datasets/datasets.py
@@ -9,9 +9,9 @@
import logging
from funasr.datasets.audio_datasets.load_audio_extract_fbank import load_audio, extract_fbank
-from funasr.utils.register import register_class, registry_tables
+from funasr.register import tables
-@register_class("dataset_classes", "AudioDataset")
+@tables.register("dataset_classes", "AudioDataset")
class AudioDataset(torch.utils.data.Dataset):
def __init__(self,
path,
@@ -22,16 +22,16 @@
float_pad_value: float = 0.0,
**kwargs):
super().__init__()
- index_ds_class = registry_tables.index_ds_classes.get(index_ds.lower())
+ index_ds_class = tables.index_ds_classes.get(index_ds.lower())
self.index_ds = index_ds_class(path)
preprocessor_speech = kwargs.get("preprocessor_speech", None)
if preprocessor_speech:
- preprocessor_speech_class = registry_tables.preprocessor_speech_classes.get(preprocessor_speech.lower())
+ preprocessor_speech_class = tables.preprocessor_speech_classes.get(preprocessor_speech.lower())
preprocessor_speech = preprocessor_speech_class(**kwargs.get("preprocessor_speech_conf"))
self.preprocessor_speech = preprocessor_speech
preprocessor_text = kwargs.get("preprocessor_text", None)
if preprocessor_text:
- preprocessor_text_class = registry_tables.preprocessor_text_classes.get(preprocessor_text.lower())
+ preprocessor_text_class = tables.preprocessor_text_classes.get(preprocessor_text.lower())
preprocessor_text = preprocessor_text_class(**kwargs.get("preprocessor_text_conf"))
self.preprocessor_text = preprocessor_text
--
Gitblit v1.9.1