From 3eda77aa4f7336c674844e7df56d028879cefdb5 Mon Sep 17 00:00:00 2001 From: pangda Date: Tue, 31 Jan 2023 11:33:15 +0000 Subject: [PATCH] [to #42322933] re-place `import torchaudio` to avoid unnecessary requirements in framework Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/11499744 --- modelscope/msdatasets/dataset_cls/dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modelscope/msdatasets/dataset_cls/dataset.py b/modelscope/msdatasets/dataset_cls/dataset.py index 49313e90..57ee8150 100644 --- a/modelscope/msdatasets/dataset_cls/dataset.py +++ b/modelscope/msdatasets/dataset_cls/dataset.py @@ -4,7 +4,6 @@ import copy import os import datasets -import torchaudio from datasets import IterableDataset from PIL import Image @@ -94,6 +93,7 @@ class NativeIterableDataset(IterableDataset): if k.endswith('Image:FILE'): ret[k + ':Object'] = Image.open(fp=ex_cache_path) if k.endswith('Audio:FILE'): + import torchaudio waveform_and_rate = torchaudio.load(ex_cache_path) ret[k + ':Object'] = waveform_and_rate entity = ret