From a1c8f0e91458f1e7153e58cc107e79de8d9f6697 Mon Sep 17 00:00:00 2001 From: Hui Zhang Date: Thu, 19 Aug 2021 03:16:07 +0000 Subject: [PATCH] fix augment --- deepspeech/frontend/augmentor/augmentation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepspeech/frontend/augmentor/augmentation.py b/deepspeech/frontend/augmentor/augmentation.py index c479958fd..50eeea991 100644 --- a/deepspeech/frontend/augmentor/augmentation.py +++ b/deepspeech/frontend/augmentor/augmentation.py @@ -97,7 +97,7 @@ class AugmentationPipeline(): ValueError: If the augmentation json config is in incorrect format". """ - SPEC_TYPES = ('specaug') + SPEC_TYPES = {'specaug'} def __init__(self, augmentation_config: str, random_seed: int=0): self._rng = np.random.RandomState(random_seed)