fix augment

3 years ago · 9de0343807
parent 9dace62581
commit 9de0343807
1 changed files with 1 additions and 1 deletions
--- 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)