diff --git a/test/test_linalg.py b/test/test_linalg.py index ed5334392739..b86c71f0ed37 100644 --- a/test/test_linalg.py +++ b/test/test_linalg.py @@ -211,7 +211,7 @@ def cholesky_test_helper(n, batchsize, device, upper): @skipCPUIfNoLapack @dtypes(torch.float32, torch.float64, torch.complex64, torch.complex128) def test_old_cholesky_batched(self, device, dtype): - from torch.testing._internal.common_utils import random_hermitian_pd_matrix + from torch.testing._internal.common_utils import random_hermitian_pd_matrix def cholesky_test_helper(n, batch_dims, upper): A = random_hermitian_pd_matrix(n, *batch_dims, dtype=dtype, device=device)