yes no update

pytorch · Dec 12, 2020 · 6eddeb2 · 6eddeb2
1 parent 89f1228
commit 6eddeb2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchaudio/datasets/yesno.py b/torchaudio/datasets/yesno.py
@@ -84,7 +84,7 @@ def __init__(self,
                 "Dataset not found. Please use `download=True` to download it."
             )
 
-        walker = sorted([str(p.stem) for p in Path(self._path).glob('*.' + self._ext_audio)])
+        walker = sorted([str(p.stem) for p in Path(self._path).glob('*' + self._ext_audio)])
         self._walker = list(walker)
 
     def __getitem__(self, n: int) -> Tuple[Tensor, int, List[int]]: