From 026956984cd8c6fe851a67785b697be121f2d9ed Mon Sep 17 00:00:00 2001 From: Adam Narozniak Date: Fri, 30 Aug 2024 11:40:43 +0200 Subject: [PATCH] Rename SizePartitioner to IdToSizeFunctionPartitioner --- ...size_partitioner.py => id_to_size_function_partitioner.py} | 4 ++-- ...tioner_test.py => id_to_size_function_partitioner_test.py} | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) rename datasets/flwr_datasets/partitioner/{size_partitioner.py => id_to_size_function_partitioner.py} (98%) rename datasets/flwr_datasets/partitioner/{size_partitioner_test.py => id_to_size_function_partitioner_test.py} (98%) 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