Rename SizePartitioner to IdToSizeFunctionPartitioner

adap · Aug 30, 2024 · 0269569 · 0269569
1 parent d02b81a
commit 0269569
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/..._datasets/partitioner/size_partitioner.py → ...tioner/id_to_size_function_partitioner.py b/..._datasets/partitioner/size_partitioner.py → ...tioner/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/...sets/partitioner/size_partitioner_test.py → ...r/id_to_size_function_partitioner_test.py b/...sets/partitioner/size_partitioner_test.py → ...r/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