diff --git a/datasets/flwr_datasets/partitioner/size_partitioner.py b/datasets/flwr_datasets/partitioner/id_to_size_function_partitioner.py
similarity index 98%
rename from datasets/flwr_datasets/partitioner/size_partitioner.py
rename to datasets/flwr_datasets/partitioner/id_to_size_function_partitioner.py
index 35937d8b9cc7..1c2ec92662fd 100644
--- a/datasets/flwr_datasets/partitioner/size_partitioner.py
+++ b/datasets/flwr_datasets/partitioner/id_to_size_function_partitioner.py
@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 # ==============================================================================
-"""SizePartitioner class."""
+"""IdToSizeFuncitonPartitioner class."""
 
 
 from typing import Callable, Dict, List, Union
@@ -23,7 +23,7 @@
 from flwr_datasets.partitioner.partitioner import Partitioner
 
 
-class SizePartitioner(Partitioner):
+class IdToSizeFunctionPartitioner(Partitioner):
     """Base class for the deterministic size partitioning based on the `partition_id`.
 
     The client with `partition_id` has the following relationship regarding the number
diff --git a/datasets/flwr_datasets/partitioner/size_partitioner_test.py b/datasets/flwr_datasets/partitioner/id_to_size_function_partitioner_test.py
similarity index 98%
rename from datasets/flwr_datasets/partitioner/size_partitioner_test.py
rename to datasets/flwr_datasets/partitioner/id_to_size_function_partitioner_test.py
index 086ca3731e58..905aa8cc9303 100644
--- a/datasets/flwr_datasets/partitioner/size_partitioner_test.py
+++ b/datasets/flwr_datasets/partitioner/id_to_size_function_partitioner_test.py
@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 # ==============================================================================
-"""SizePartitioner tests."""
+"""IdToSizeFuncitonPartitioner tests."""
 
 
 import unittest