Unverified Commit 6bad3a66 authored by Ankit Dobhal's avatar Ankit Dobhal Committed by GitHub
Browse files

Lint code style and remove PY2 compatibility (#1386)

parent f06074aa
...@@ -35,7 +35,7 @@ def compute_spectral_flatness(frame, epsilon=0.01): ...@@ -35,7 +35,7 @@ def compute_spectral_flatness(frame, epsilon=0.01):
return -10 * torch.log10(epsilon + geometric_mean / arithmetic_mean) return -10 * torch.log10(epsilon + geometric_mean / arithmetic_mean)
class VoiceActivityDetection(object): class VoiceActivityDetection:
def __init__( def __init__(
self, self,
num_init_frames=30, num_init_frames=30,
...@@ -148,7 +148,7 @@ class VoiceActivityDetection(object): ...@@ -148,7 +148,7 @@ class VoiceActivityDetection(object):
return self.silence_mark return self.silence_mark
class MicrophoneStream(object): class MicrophoneStream:
"""Opens a recording stream as a generator yielding the audio chunks.""" """Opens a recording stream as a generator yielding the audio chunks."""
def __init__(self, device=None, rate=22050, chunk=2205): def __init__(self, device=None, rate=22050, chunk=2205):
......
...@@ -8,7 +8,7 @@ class LanguageModel: ...@@ -8,7 +8,7 @@ class LanguageModel:
self.char_space = char_space self.char_space = char_space
self.char_blank = char_blank self.char_blank = char_blank
labels = [l for l in labels] labels = list(labels)
self.length = len(labels) self.length = len(labels)
enumerated = list(enumerate(labels)) enumerated = list(enumerate(labels))
flipped = [(sub[1], sub[0]) for sub in enumerated] flipped = [(sub[1], sub[0]) for sub in enumerated]
......
...@@ -156,7 +156,7 @@ class Test_Kaldi(common_utils.TempDirMixin, common_utils.TorchaudioTestCase): ...@@ -156,7 +156,7 @@ class Test_Kaldi(common_utils.TempDirMixin, common_utils.TorchaudioTestCase):
# Read kaldi's output from file # Read kaldi's output from file
kaldi_output_path = os.path.join(self.kaldi_output_dir, f) kaldi_output_path = os.path.join(self.kaldi_output_dir, f)
kaldi_output_dict = {k: v for k, v in torchaudio.kaldi_io.read_mat_ark(kaldi_output_path)} kaldi_output_dict = dict(torchaudio.kaldi_io.read_mat_ark(kaldi_output_path))
assert len(kaldi_output_dict) == 1 and 'my_id' in kaldi_output_dict, 'invalid test kaldi ark file' assert len(kaldi_output_dict) == 1 and 'my_id' in kaldi_output_dict, 'invalid test kaldi ark file'
kaldi_output = kaldi_output_dict['my_id'] kaldi_output = kaldi_output_dict['my_id']
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment