Update component module functions docstrings for fix lint warnings

openfl/component/__init__.py

@@ -1,24 +1,16 @@
 # Copyright (C) 2020-2023 Intel Corporation
 # SPDX-License-Identifier: Apache-2.0
-
 """openfl.component package."""
 
 from .aggregator import Aggregator
-from .assigner import Assigner
-from .assigner import RandomGroupedAssigner
-from .assigner import StaticGroupedAssigner
+from .assigner import Assigner, RandomGroupedAssigner, StaticGroupedAssigner
 from .collaborator import Collaborator
-from .straggler_handling_functions import StragglerHandlingFunction
-from .straggler_handling_functions import CutoffTimeBasedStragglerHandling
-from .straggler_handling_functions import PercentageBasedStragglerHandling
+from .straggler_handling_functions import (CutoffTimeBasedStragglerHandling,
+                                           PercentageBasedStragglerHandling,
+                                           StragglerHandlingFunction)
 
 __all__ = [
-    'Assigner',
-    'RandomGroupedAssigner',
-    'StaticGroupedAssigner',
-    'Aggregator',
-    'Collaborator',
-    'StragglerHandlingFunction',
-    'CutoffTimeBasedStragglerHandling',
-    'PercentageBasedStragglerHandling'
+    'Assigner', 'RandomGroupedAssigner', 'StaticGroupedAssigner', 'Aggregator',
+    'Collaborator', 'StragglerHandlingFunction',
+    'CutoffTimeBasedStragglerHandling', 'PercentageBasedStragglerHandling'
 ]

openfl/component/assigner/random_grouped_assigner.py

@@ -54,10 +54,9 @@ def define_task_assignments(self):
         Returns:
             None
         """
-        assert (
-            np.abs(1.0 -
-                   np.sum([group['percentage'] for group in self.task_groups]))
-            < 0.01), ('Task group percentages must sum to 100%')
+        assert (np.abs(1.0 - np.sum([group['percentage']
+                                     for group in self.task_groups])) < 0.01), (
+            'Task group percentages must sum to 100%')
 
         # Start by finding all of the tasks in all specified groups
         self.all_tasks_in_groups = list(
@@ -83,9 +82,8 @@ def define_task_assignments(self):
             for group in self.task_groups:
                 num_col_in_group = int(group['percentage'] * col_list_size)
                 rand_col_group_list = [
-                    self.authorized_cols[i]
-                    for i in randomized_col_idx[col_idx:col_idx +
-                                                num_col_in_group]
+                    self.authorized_cols[i] for i in
+                    randomized_col_idx[col_idx:col_idx + num_col_in_group]
                 ]
                 self.task_group_collaborators[
                     group['name']] = rand_col_group_list

openfl/component/collaborator/__init__.py

@@ -1,6 +1,5 @@
 # Copyright (C) 2020-2023 Intel Corporation
 # SPDX-License-Identifier: Apache-2.0
-
 """Collaborator package."""
 
 from .collaborator import Collaborator

openfl/component/director/__init__.py

@@ -1,11 +1,9 @@
 # Copyright (C) 2020-2023 Intel Corporation
 # SPDX-License-Identifier: Apache-2.0
-
 """Director package."""
 
 from .director import Director
 
-
 __all__ = [
     'Director',
 ]