Sort imports

datasets/flwr_datasets/partitioner/__init__.py

@@ -22,10 +22,10 @@
 from .linear_partitioner import LinearPartitioner
 from .natural_id_partitioner import NaturalIdPartitioner
 from .partitioner import Partitioner
+from .semantic_partitioner import SemanticPartitioner
 from .shard_partitioner import ShardPartitioner
 from .size_partitioner import SizePartitioner
 from .square_partitioner import SquarePartitioner
-from .semantic_partitioner import SemanticPartitioner
 
 __all__ = [
     "DirichletPartitioner",
@@ -35,8 +35,8 @@
     "LinearPartitioner",
     "NaturalIdPartitioner",
     "Partitioner",
+    "SemanticPartitioner"
     "ShardPartitioner",
     "SizePartitioner",
     "SquarePartitioner",
-    "SemanticPartitioner"
 ]

datasets/flwr_datasets/partitioner/semantic_partitioner.py

@@ -18,16 +18,16 @@
 import warnings
 from typing import Any, Callable, Dict, List, Optional, Union
 
-import torch
 import numpy as np
-import datasets
-from torch.distributions import MultivariateNormal, kl_divergence
-from torchvision import models
+import torch
 from scipy.optimize import linear_sum_assignment
-from sklearn.preprocessing import StandardScaler
 from sklearn.decomposition import PCA
 from sklearn.mixture import GaussianMixture
+from sklearn.preprocessing import StandardScaler
+from torch.distributions import MultivariateNormal, kl_divergence
+from torchvision import models
 
+import datasets
 from flwr_datasets.common.typing import NDArrayFloat
 from flwr_datasets.partitioner.partitioner import Partitioner
 

datasets/flwr_datasets/partitioner/semantic_partitioner_test.py

@@ -21,10 +21,10 @@
 from typing import Tuple
 
 import numpy as np
-from datasets import Dataset
-from torchvision import models
 from parameterized import parameterized
+from torchvision import models
 
+from datasets import Dataset
 from flwr_datasets.partitioner.semantic_partitioner import SemanticPartitioner