Skip to content

Commit

Permalink
flake8 fix
Browse files Browse the repository at this point in the history
  • Loading branch information
IvanYashchuk committed Nov 13, 2020
1 parent 88e23c3 commit eb507e5
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion test/test_linalg.py
Expand Up @@ -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)
Expand Down

0 comments on commit eb507e5

Please sign in to comment.