GokseninYuksel commited on
Commit
483ffb9
·
verified ·
1 Parent(s): cb8de63

Update feature_extraction_gramt_ambisonics.py

Browse files
feature_extraction_gramt_ambisonics.py CHANGED
@@ -7,7 +7,7 @@ from transformers.utils import TensorType
7
 
8
  import torch
9
  from torch import Tensor
10
- from typing import Callable
11
  from torchaudio.transforms import Spectrogram, MelScale
12
 
13
  class FeatureExtractor(torch.nn.Module):
@@ -192,7 +192,7 @@ class AmbisonicsFeatureExtractor(SequenceFeatureExtractor):
192
 
193
  def __call__(
194
  self,
195
- raw_speech: Union[np.ndarray, list[float], list[np.ndarray], list[list[float]]],
196
  sampling_rate: Optional[int] = None,
197
  return_tensors: Optional[Union[str, TensorType]] = None,
198
  **kwargs,
 
7
 
8
  import torch
9
  from torch import Tensor
10
+ from typing import Callable, List
11
  from torchaudio.transforms import Spectrogram, MelScale
12
 
13
  class FeatureExtractor(torch.nn.Module):
 
192
 
193
  def __call__(
194
  self,
195
+ raw_speech: Union[np.ndarray, List[float], List[np.ndarray], List[List[float]]],
196
  sampling_rate: Optional[int] = None,
197
  return_tensors: Optional[Union[str, TensorType]] = None,
198
  **kwargs,