diff --git a/datasets/flwr_datasets/partitioner/__init__.py b/datasets/flwr_datasets/partitioner/__init__.py
index 3d2b59c15475..3fed4446db42 100644
--- a/datasets/flwr_datasets/partitioner/__init__.py
+++ b/datasets/flwr_datasets/partitioner/__init__.py
@@ -19,7 +19,7 @@
 from .distribution_partitioner import DistributionPartitioner
 from .exponential_partitioner import ExponentialPartitioner
 from .grouped_natural_id_partitioner import GroupedNaturalIdPartitioner
-from .id_to_size_function_partitioner import IdToSizeFunctionPartitioner
+from .id_to_size_fnc_partitioner import IdToSizeFncPartitioner
 from .iid_partitioner import IidPartitioner
 from .inner_dirichlet_partitioner import InnerDirichletPartitioner
 from .linear_partitioner import LinearPartitioner
@@ -34,7 +34,7 @@
     "DistributionPartitioner",
     "ExponentialPartitioner",
     "GroupedNaturalIdPartitioner",
-    "IdToSizeFunctionPartitioner",
+    "IdToSizeFncPartitioner",
     "IidPartitioner",
     "InnerDirichletPartitioner",
     "LinearPartitioner",
diff --git a/datasets/flwr_datasets/partitioner/exponential_partitioner.py b/datasets/flwr_datasets/partitioner/exponential_partitioner.py
index 86d26c03c82f..1bf838df5909 100644
--- a/datasets/flwr_datasets/partitioner/exponential_partitioner.py
+++ b/datasets/flwr_datasets/partitioner/exponential_partitioner.py
@@ -17,12 +17,10 @@
 
 import numpy as np
 
-from flwr_datasets.partitioner.id_to_size_function_partitioner import (
-    IdToSizeFunctionPartitioner,
-)
+from flwr_datasets.partitioner.id_to_size_fnc_partitioner import IdToSizeFncPartitioner
 
 
-class ExponentialPartitioner(IdToSizeFunctionPartitioner):
+class ExponentialPartitioner(IdToSizeFncPartitioner):
     """Partitioner creates partitions of size that are correlated with exp(id).
 
     The amount of data each client gets is correlated with the exponent of partition ID.
diff --git a/datasets/flwr_datasets/partitioner/id_to_size_function_partitioner.py b/datasets/flwr_datasets/partitioner/id_to_size_fnc_partitioner.py
similarity index 98%
rename from datasets/flwr_datasets/partitioner/id_to_size_function_partitioner.py
rename to datasets/flwr_datasets/partitioner/id_to_size_fnc_partitioner.py
index 1c2ec92662fd..d7b3b21037de 100644
--- a/datasets/flwr_datasets/partitioner/id_to_size_function_partitioner.py
+++ b/datasets/flwr_datasets/partitioner/id_to_size_fnc_partitioner.py
@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 # ==============================================================================
-"""IdToSizeFuncitonPartitioner class."""
+"""IdToSizeFncPartitioner class."""
 
 
 from typing import Callable, Dict, List, Union
@@ -23,7 +23,7 @@
 from flwr_datasets.partitioner.partitioner import Partitioner
 
 
-class IdToSizeFunctionPartitioner(Partitioner):
+class IdToSizeFncPartitioner(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/id_to_size_function_partitioner_test.py b/datasets/flwr_datasets/partitioner/id_to_size_fnc_partitioner_test.py
similarity index 100%
rename from datasets/flwr_datasets/partitioner/id_to_size_function_partitioner_test.py
rename to datasets/flwr_datasets/partitioner/id_to_size_fnc_partitioner_test.py
diff --git a/datasets/flwr_datasets/partitioner/linear_partitioner.py b/datasets/flwr_datasets/partitioner/linear_partitioner.py
index dab4ad57c5ac..07fea16df5e0 100644
--- a/datasets/flwr_datasets/partitioner/linear_partitioner.py
+++ b/datasets/flwr_datasets/partitioner/linear_partitioner.py
@@ -15,12 +15,10 @@
 """LinearPartitioner class."""
 
 
-from flwr_datasets.partitioner.id_to_size_function_partitioner import (
-    IdToSizeFunctionPartitioner,
-)
+from flwr_datasets.partitioner.id_to_size_fnc_partitioner import IdToSizeFncPartitioner
 
 
-class LinearPartitioner(IdToSizeFunctionPartitioner):
+class LinearPartitioner(IdToSizeFncPartitioner):
     """Partitioner creates partitions of size that are linearly correlated with id.
 
     The amount of data each client gets is linearly correlated with the partition ID.
diff --git a/datasets/flwr_datasets/partitioner/square_partitioner.py b/datasets/flwr_datasets/partitioner/square_partitioner.py
index cb3950ea6aec..d48af247e5cb 100644
--- a/datasets/flwr_datasets/partitioner/square_partitioner.py
+++ b/datasets/flwr_datasets/partitioner/square_partitioner.py
@@ -17,12 +17,10 @@
 
 import numpy as np
 
-from flwr_datasets.partitioner.id_to_size_function_partitioner import (
-    IdToSizeFunctionPartitioner,
-)
+from flwr_datasets.partitioner.id_to_size_fnc_partitioner import IdToSizeFncPartitioner
 
 
-class SquarePartitioner(IdToSizeFunctionPartitioner):
+class SquarePartitioner(IdToSizeFncPartitioner):
     """Partitioner creates partitions of size that are correlated with squared id.
 
     The amount of data each client gets is correlated with the squared partition ID.