Run isort

falkon/kernels/distance_kernel.py

@@ -1,4 +1,4 @@
-from typing import Dict, Optional, Type, Union, Tuple
+from typing import Dict, Optional, Tuple, Type, Union
 
 import numpy as np
 import torch

falkon/kernels/keops_helpers.py

@@ -1,6 +1,6 @@
 import abc
 import functools
-from typing import List, Optional, Union, Dict
+from typing import Dict, List, Optional, Union
 
 import torch
 

falkon/mmv_ops/fmm.py

@@ -1,18 +1,12 @@
 from contextlib import ExitStack
 from dataclasses import dataclass, field
-from typing import Optional, Union, Dict
+from typing import Dict, Optional, Union
 
 import torch
 import torch.cuda as tcd
 
 import falkon
-from falkon.mmv_ops.utils import (
-    _call_direct,
-    _check_contiguity,
-    _extract_flat,
-    _get_gpu_info,
-    _start_wait_processes,
-)
+from falkon.mmv_ops.utils import _call_direct, _check_contiguity, _extract_flat, _get_gpu_info, _start_wait_processes
 from falkon.options import BaseOptions
 from falkon.sparse.sparse_tensor import SparseTensor
 from falkon.utils.device_copy import copy

falkon/mmv_ops/fmmv.py

@@ -1,7 +1,7 @@
 from collections import defaultdict
 from contextlib import ExitStack
 from dataclasses import dataclass, field
-from typing import Optional, Tuple, Union, Dict
+from typing import Dict, Optional, Tuple, Union
 
 import numpy as np
 import torch
@@ -22,12 +22,7 @@
 from falkon.options import BaseOptions
 from falkon.sparse import SparseTensor
 from falkon.utils.device_copy import copy
-from falkon.utils.helpers import (
-    calc_gpu_block_sizes,
-    select_dim_over_n,
-    select_dim_over_nm_v2,
-    sizeof_dtype,
-)
+from falkon.utils.helpers import calc_gpu_block_sizes, select_dim_over_n, select_dim_over_nm_v2, sizeof_dtype
 from falkon.utils.tensor_helpers import create_same_stride, extract_fortran