From 42277221650c60dd4f3b4d4bcea4c671e2ef0433 Mon Sep 17 00:00:00 2001
From: 嘉渊 <wangjiaming.wjm@alibaba-inc.com>
Date: 星期四, 18 五月 2023 15:32:23 +0800
Subject: [PATCH] update repo
---
funasr/datasets/small_datasets/dataset.py | 10 +++++-----
1 files changed, 5 insertions(+), 5 deletions(-)
diff --git a/funasr/datasets/small_datasets/dataset.py b/funasr/datasets/small_datasets/dataset.py
index 7689a02..123f109 100644
--- a/funasr/datasets/small_datasets/dataset.py
+++ b/funasr/datasets/small_datasets/dataset.py
@@ -173,8 +173,8 @@
raise RuntimeError(f"{k} is duplicated ({path}:{linenum})")
text_loader[k] = v
return text_loader
- elif loader_type == "text_in":
- text_in_loader = {}
+ elif loader_type == "text_int":
+ text_int_loader = {}
with open(path, "r", encoding="utf-8") as f:
for linenum, line in enumerate(f, 1):
sps = line.rstrip().split(maxsplit=1)
@@ -182,10 +182,10 @@
k, v = sps[0], ""
else:
k, v = sps
- if k in text_in_loader:
+ if k in text_int_loader:
raise RuntimeError(f"{k} is duplicated ({path}:{linenum})")
- text_in_loader[k] = [int(i) for i in v.split()]
- return text_in_loader
+ text_int_loader[k] = [int(i) for i in v.split()]
+ return text_int_loader
else:
raise RuntimeError(f"Not supported: loader_type={loader_type}")
--
Gitblit v1.9.1