diff --git a/src/beignet/_chebyshev_polynomial_power.py b/src/beignet/_chebyshev_polynomial_power.py index ab135f6d82..8f1c973818 100644 --- a/src/beignet/_chebyshev_polynomial_power.py +++ b/src/beignet/_chebyshev_polynomial_power.py @@ -2,7 +2,7 @@ import torch from torch import Tensor -from torchaudio.functional import convolve +from ._convolve import convolve from ._add_chebyshev_polynomial import add_chebyshev_polynomial diff --git a/src/beignet/_multiply_chebyshev_polynomial.py b/src/beignet/_multiply_chebyshev_polynomial.py index 04729410df..299d56d10a 100644 --- a/src/beignet/_multiply_chebyshev_polynomial.py +++ b/src/beignet/_multiply_chebyshev_polynomial.py @@ -3,7 +3,7 @@ import torch from torch import Tensor -from torchaudio.functional import convolve +from ._convolve import convolve def multiply_chebyshev_polynomial( diff --git a/src/beignet/_multiply_polynomial.py b/src/beignet/_multiply_polynomial.py index ea438b1d2c..5a6cbe7530 100644 --- a/src/beignet/_multiply_polynomial.py +++ b/src/beignet/_multiply_polynomial.py @@ -2,7 +2,7 @@ import torch from torch import Tensor -from torchaudio.functional import convolve +from ._convolve import convolve def multiply_polynomial(