Skip to content

GitLab

Commit 1ef6de08 authored by Jason Lian's avatar Jason Lian
Browse files

more

parent 2a716468
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
build_tools/travis/install.sh
View file @ 1ef6de08
...@@ -53,7 +53,7 @@ pip install -r requirements.txt ...@@ -53,7 +53,7 @@ pip install -r requirements.txt
# Install the following only if running tests # Install the following only if running tests
if [[ "$SKIP_TESTS" != "true" ]]; then if [[ "$SKIP_TESTS" != "true" ]]; then
# PyTorch (nightly as 1.1 does not have Optional for type annotations) # PyTorch (nightly as 1.1 does not have Optional for type annotations)
conda install --yes pytorch-nightly-cpu=1.1.0.dev20190516 -c pytorch conda install --yes pytorch-nightly-cpu -c pytorch
# TorchAudio CPP Extensions # TorchAudio CPP Extensions
pip install . pip install .
......
torchaudio/functional.py
View file @ 1ef6de08
...@@ -102,6 +102,7 @@ def LC2CL(tensor): ...@@ -102,6 +102,7 @@ def LC2CL(tensor):
return tensor.transpose(0, 1).contiguous() return tensor.transpose(0, 1).contiguous()
@torch.jit.ignore
def _stft(input, n_fft, hop_length, win_length, window, center, pad_mode, normalized, onesided): def _stft(input, n_fft, hop_length, win_length, window, center, pad_mode, normalized, onesided):
# type: (Tensor, int, Optional[int], Optional[int], Optional[Tensor], bool, str, bool, bool) -> Tensor # type: (Tensor, int, Optional[int], Optional[int], Optional[Tensor], bool, str, bool, bool) -> Tensor
return torch.stft(input, n_fft, hop_length, win_length, window, center, pad_mode, normalized, onesided) return torch.stft(input, n_fft, hop_length, win_length, window, center, pad_mode, normalized, onesided)
......
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