Backward compatible use of RegEx match objects

mozilla · May 8, 2020 · cf84a32 · cf84a32
1 parent fe2bb68
commit cf84a32
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/training/deepspeech_training/util/signal_augmentations.py b/training/deepspeech_training/util/signal_augmentations.py
@@ -163,11 +163,11 @@ def parse_augmentation(augmentation_spec):
     match = SPEC_PARSER.match(augmentation_spec)
     if not match:
         raise ValueError('Augmentation specification has wrong format')
-    cls_name = match[1][0].upper() + match[1][1:]
+    cls_name = match.group(1)[0].upper() + match.group(1)[1:]
     if cls_name not in globals():
         raise ValueError('Unknown augmentation: {}'.format(cls_name))
     augmentation_cls = globals()[cls_name]
-    parameters = [] if match[3] is None else match[3].split(',')
+    parameters = [] if match.group(3) is None else match.group(3).split(',')
     args = []
     kwargs = {}
     for parameter in parameters: