diff --git a/test/norm/test_hutchinson.py b/test/norm/test_hutchinson.py index 0ff5cd1..18fdc9e 100644 --- a/test/norm/test_hutchinson.py +++ b/test/norm/test_hutchinson.py @@ -3,7 +3,6 @@ from functools import partial from test.utils import check_estimator_convergence -from numpy import mean from numpy.random import rand, seed from pytest import mark diff --git a/test/utils.py b/test/utils.py index 1168386..69bfe78 100644 --- a/test/utils.py +++ b/test/utils.py @@ -7,7 +7,7 @@ from einops import rearrange, reduce from einops.layers.torch import Rearrange -from numpy import eye, mean, ndarray +from numpy import eye, ndarray from torch import Tensor, allclose, as_tensor, cat, cuda, device, dtype from torch import eye as torch_eye from torch import from_numpy, rand, randint, zeros_like