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,
|
| 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,
|