diff --git a/datasets/flwr_datasets/partitioner/distribution_partitioner.py b/datasets/flwr_datasets/partitioner/distribution_partitioner.py
index dfbe2d6821ac..b4876bdd70f9 100644
--- a/datasets/flwr_datasets/partitioner/distribution_partitioner.py
+++ b/datasets/flwr_datasets/partitioner/distribution_partitioner.py
@@ -17,9 +17,9 @@
 from typing import Dict, List, Optional
 
 import numpy as np
-from flwr_datasets.partitioner.partitioner import Partitioner
 
 import datasets
+from flwr_datasets.partitioner.partitioner import Partitioner
 
 
 class DistributionPartitioner(Partitioner):  # pylint: disable=R0902