Skip to content

Commit e6c1151

Browse files
committed
Undo drop support for forced_aling and rnnt_loss
1 parent dca045a commit e6c1151

File tree

2 files changed

+2
-12
lines changed

2 files changed

+2
-12
lines changed

src/torchaudio/functional/__init__.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,4 @@
1-
from torchaudio._internal.module_utils import dropping_support
2-
3-
from ._alignment import forced_align as _forced_align, merge_tokens, TokenSpan
1+
from ._alignment import forced_align, merge_tokens, TokenSpan
42
from .filtering import (
53
allpass_biquad,
64
band_biquad,
@@ -26,8 +24,6 @@
2624
vad,
2725
)
2826

29-
forced_align = dropping_support(_forced_align)
30-
3127
from .functional import (
3228
add_noise,
3329
amplitude_to_DB,

src/torchaudio/functional/functional.py

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,6 @@
88
import torch
99
import torchaudio
1010
from torch import Tensor
11-
from torchaudio._internal.module_utils import dropping_support
1211

1312
from .filtering import highpass_biquad, treble_biquad
1413

@@ -1733,7 +1732,7 @@ def backward(ctx, dy):
17331732
return (result, None, None, None, None, None, None, None)
17341733

17351734

1736-
def _rnnt_loss(
1735+
def rnnt_loss(
17371736
logits: Tensor,
17381737
targets: Tensor,
17391738
logit_lengths: Tensor,
@@ -1830,11 +1829,6 @@ def psd(
18301829
return psd
18311830

18321831

1833-
# Expose both deprecated wrapper as well as original because torchscript breaks on
1834-
# wrapped functions.
1835-
rnnt_loss = dropping_support(_rnnt_loss)
1836-
1837-
18381832
def _compute_mat_trace(input: torch.Tensor, dim1: int = -1, dim2: int = -2) -> torch.Tensor:
18391833
r"""Compute the trace of a Tensor along ``dim1`` and ``dim2`` dimensions.
18401834

0 commit comments

Comments
 (0)