"src/vscode:/vscode.git/clone" did not exist on "d5dd8df3b4e978e3f0549fa35e245a8785116af7"
Unverified Commit 5c9d33de authored by cpuhrsch's avatar cpuhrsch Committed by GitHub
Browse files

Hotfix due to PyTorch JIT issue (#149)

parents 4beb982f 5033fdd0
...@@ -9,7 +9,7 @@ python --version ...@@ -9,7 +9,7 @@ python --version
run_tests() { run_tests() {
# find all the test files that match "test*.py" # find all the test files that match "test*.py"
TEST_FILES="$(find test/ -type f -name "test*.py" | sort)" TEST_FILES="$(find test -type f -name "test*.py" | sort)"
echo "Test files are:" echo "Test files are:"
echo $TEST_FILES echo $TEST_FILES
......
...@@ -101,7 +101,8 @@ def LC2CL(tensor): ...@@ -101,7 +101,8 @@ def LC2CL(tensor):
""" """
return tensor.transpose(0, 1).contiguous() return tensor.transpose(0, 1).contiguous()
# TODO: remove this once https://github.com/pytorch/pytorch/issues/21478 gets solved
@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