diff --git a/falkon/kernels/precomputed_kernel.py b/falkon/kernels/precomputed_kernel.py index 6583ffa..0f9bb7f 100644 --- a/falkon/kernels/precomputed_kernel.py +++ b/falkon/kernels/precomputed_kernel.py @@ -3,9 +3,9 @@ import torch from torch import Tensor -from falkon import FalkonOptions from falkon.kernels import Kernel from falkon.mmv_ops.fmmv_incore import incore_fdmmv, incore_fmmv +from falkon.options import FalkonOptions from falkon.sparse import SparseTensor from falkon.utils.helpers import check_same_dtype