From f178effcb9309e4ef3667b5173ba8afbad014ea0 Mon Sep 17 00:00:00 2001
From: KarhouTam <karhoutam@qq.com>
Date: Fri, 21 Jun 2024 00:22:12 +0800
Subject: [PATCH] Sort imports

---
 datasets/flwr_datasets/partitioner/__init__.py         |  4 ++--
 .../flwr_datasets/partitioner/semantic_partitioner.py  | 10 +++++-----
 .../partitioner/semantic_partitioner_test.py           |  4 ++--
 3 files changed, 9 insertions(+), 9 deletions(-)

diff --git a/datasets/flwr_datasets/partitioner/__init__.py b/datasets/flwr_datasets/partitioner/__init__.py
index 3a85c195707a..72e696a606c6 100644
--- a/datasets/flwr_datasets/partitioner/__init__.py
+++ b/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"
 ]
diff --git a/datasets/flwr_datasets/partitioner/semantic_partitioner.py b/datasets/flwr_datasets/partitioner/semantic_partitioner.py
index f9fca53b0a3e..f43a38e148d6 100644
--- a/datasets/flwr_datasets/partitioner/semantic_partitioner.py
+++ b/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
 
diff --git a/datasets/flwr_datasets/partitioner/semantic_partitioner_test.py b/datasets/flwr_datasets/partitioner/semantic_partitioner_test.py
index f6b06cd8f75c..730fd8e7545e 100644
--- a/datasets/flwr_datasets/partitioner/semantic_partitioner_test.py
+++ b/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