fix: return type annotation

DiffAPF · Apr 19, 2024 · 8bf7bd5 · 8bf7bd5
1 parent 79c3a98
commit 8bf7bd5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchlpc/core.py b/torchlpc/core.py
@@ -137,7 +137,7 @@ def lpc_cuda(x: torch.Tensor, A: torch.Tensor, zi: torch.Tensor) -> torch.Tensor
 
 
 @njit(parallel=True)
-def lpc_np(x: np.ndarray, A: np.ndarray, zi: np.ndarray) -> None:
+def lpc_np(x: np.ndarray, A: np.ndarray, zi: np.ndarray) -> np.ndarray:
     B, T = x.shape
     order = zi.shape[1]
     padded_y = np.empty((B, T + order), dtype=x.dtype)