small bugfix

RF5 · Nov 30, 2021 · a474a57 · a474a57
1 parent 5a9586d
commit a474a57
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ge2e_embedding.py b/ge2e_embedding.py
@@ -26,9 +26,9 @@ def melspec_from_file(self, x):
         """ Convert file path `x` to a [n_frames, n_mels] shape mel-spectrogram tensor """
         return self.melspec_tfm.from_file(x).T
 
-    def melspec_from_array(self, x):
+    def melspec_from_array(self, x, sr):
         """ Convert a 1D torch tensor in to a [n_frames, n_mels] shape mel-spectrogram tensor """
-        return self.melspec_tfm.from_array(x).T
+        return self.melspec_tfm.from_array(x, sr).T
 
     def print_hparams(self):
         for key in self.hparams.__dict__():