removed torchaudio

Genentech · Oct 3, 2024 · 1d6dc81 · 1d6dc81
1 parent bde1b44
commit 1d6dc81
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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
@@ -2,7 +2,7 @@
 
 import torch
 from torch import Tensor
-from torchaudio.functional import convolve
+from ._convolve import convolve
 
 
 def multiply_polynomial(