[DEL] Remove unused imports

f-dangel · Nov 8, 2023 · 2a04796 · 2a04796
1 parent 1e52ec1
commit 2a04796
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/curvlinops/kfac.py b/curvlinops/kfac.py
@@ -19,12 +19,8 @@
 
 from einops import rearrange
 from numpy import ndarray
-from torch import Generator, Tensor, cat, einsum
-from torch import mean as torch_mean
-from torch import no_grad, randn, stack
-from torch import sum as torch_sum
+from torch import Generator, Tensor, cat, einsum, randn, stack
 from torch.nn import CrossEntropyLoss, Linear, Module, MSELoss, Parameter
-from torch.nn.functional import log_softmax, softmax
 from torch.utils.hooks import RemovableHandle
 
 from curvlinops._base import _LinearOperator