From 614c6fc304a330cddedc9df3b410f99dfe4348d9 Mon Sep 17 00:00:00 2001
From: Adam Narozniak <adam@flower.dev>
Date: Tue, 24 Oct 2023 15:39:45 +0200
Subject: [PATCH] Add new partitioners to init for easy imports

---
 datasets/flwr_datasets/partitioner/__init__.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/datasets/flwr_datasets/partitioner/__init__.py b/datasets/flwr_datasets/partitioner/__init__.py
index d1eba7895702..83186a6d24ee 100644
--- a/datasets/flwr_datasets/partitioner/__init__.py
+++ b/datasets/flwr_datasets/partitioner/__init__.py
@@ -15,10 +15,18 @@
 """Flower Datasets Partitioner package."""
 
 
+from .exponential_partitioner import ExponentialPartitioner
 from .iid_partitioner import IidPartitioner
+from .linear_partitioner import LinearPartitioner
 from .partitioner import Partitioner
+from .size_partitioner import SizePartitioner
+from .square_partitioner import SquarePartitioner
 
 __all__ = [
     "IidPartitioner",
     "Partitioner",
+    "SizePartitioner",
+    "LinearPartitioner",
+    "SquarePartitioner",
+    "ExponentialPartitioner",
 ]