From 2606ed6ffb1396376825dcce2d30bd89ce0a659d Mon Sep 17 00:00:00 2001
From: Ishant Thakare <ishantrog752@gmail.com>
Date: Fri, 18 Oct 2024 12:15:28 +0530
Subject: [PATCH 1/6] Renamed Straggler Handling package

Signed-off-by: Ishant Thakare <ishantrog752@gmail.com>
---
 .../straggler_handling_algorithms.rst         |   2 +-
 docs/source/api/openfl_component.rst          |   2 +-
 .../plan/plan.yaml                            |   2 +-
 openfl/component/__init__.py                  |   6 +-
 openfl/component/aggregator/aggregator.py     |   2 +-
 .../straggler_handling_functions/__init__.py  |  13 --
 .../percentage_based_straggler_handling.py    |  78 -----------
 .../straggler_handling_function.py            |  59 --------
 .../straggler_handling_policy/__init__.py     |   9 ++
 .../straggler_handling_policy.py}             | 126 +++++++++++++++++-
 openfl/federated/plan/plan.py                 |   2 +-
 setup.py                                      |   2 +-
 12 files changed, 138 insertions(+), 165 deletions(-)
 delete mode 100644 openfl/component/straggler_handling_functions/__init__.py
 delete mode 100644 openfl/component/straggler_handling_functions/percentage_based_straggler_handling.py
 delete mode 100644 openfl/component/straggler_handling_functions/straggler_handling_function.py
 create mode 100644 openfl/component/straggler_handling_policy/__init__.py
 rename openfl/component/{straggler_handling_functions/cutoff_time_based_straggler_handling.py => straggler_handling_policy/straggler_handling_policy.py} (54%)

diff --git a/docs/developer_guide/advanced_topics/straggler_handling_algorithms.rst b/docs/developer_guide/advanced_topics/straggler_handling_algorithms.rst
index a71b1385e5..d5f00500e0 100644
--- a/docs/developer_guide/advanced_topics/straggler_handling_algorithms.rst
+++ b/docs/developer_guide/advanced_topics/straggler_handling_algorithms.rst
@@ -34,7 +34,7 @@ The example template, **torch_cnn_mnist_straggler_check**, uses the ``Percentage
     .. code-block:: yaml
     
         straggler_handling_policy :
-            template : openfl.component.straggler_handling_functions.CutoffTimeBasedStragglerHandling
+            template : openfl.component.straggler_handling_policy.CutoffTimeBasedStragglerHandling
             settings :
                 straggler_cutoff_time : 20
                 minimum_reporting : 1
diff --git a/docs/source/api/openfl_component.rst b/docs/source/api/openfl_component.rst
index 8deb6528ac..67639599f5 100644
--- a/docs/source/api/openfl_component.rst
+++ b/docs/source/api/openfl_component.rst
@@ -17,4 +17,4 @@ Component modules reference:
    openfl.component.collaborator
    openfl.component.director
    openfl.component.envoy
-   openfl.component.straggler_handling_functions
+   openfl.component.straggler_handling_policy
diff --git a/openfl-workspace/torch_cnn_mnist_straggler_check/plan/plan.yaml b/openfl-workspace/torch_cnn_mnist_straggler_check/plan/plan.yaml
index a42b064e56..cca9414fb6 100644
--- a/openfl-workspace/torch_cnn_mnist_straggler_check/plan/plan.yaml
+++ b/openfl-workspace/torch_cnn_mnist_straggler_check/plan/plan.yaml
@@ -45,7 +45,7 @@ compression_pipeline :
   defaults : plan/defaults/compression_pipeline.yaml
 
 straggler_handling_policy :
-  template : openfl.component.straggler_handling_functions.PercentageBasedStragglerHandling
+  template : openfl.component.straggler_handling_policy.CutoffTimeBasedStragglerHandling
   settings : 
     percent_collaborators_needed : 0.5
     minimum_reporting : 1
\ No newline at end of file
diff --git a/openfl/component/__init__.py b/openfl/component/__init__.py
index 3b787f87d0..97788f4a6f 100644
--- a/openfl/component/__init__.py
+++ b/openfl/component/__init__.py
@@ -7,12 +7,8 @@
 from openfl.component.assigner.random_grouped_assigner import RandomGroupedAssigner
 from openfl.component.assigner.static_grouped_assigner import StaticGroupedAssigner
 from openfl.component.collaborator.collaborator import Collaborator
-from openfl.component.straggler_handling_functions.cutoff_time_based_straggler_handling import (
+from openfl.component.straggler_handling_policy import (
     CutoffTimeBasedStragglerHandling,
-)
-from openfl.component.straggler_handling_functions.percentage_based_straggler_handling import (
     PercentageBasedStragglerHandling,
-)
-from openfl.component.straggler_handling_functions.straggler_handling_function import (
     StragglerHandlingPolicy,
 )
diff --git a/openfl/component/aggregator/aggregator.py b/openfl/component/aggregator/aggregator.py
index 81d3e7411a..3540c0791e 100644
--- a/openfl/component/aggregator/aggregator.py
+++ b/openfl/component/aggregator/aggregator.py
@@ -8,7 +8,7 @@
 from logging import getLogger
 from threading import Lock
 
-from openfl.component.straggler_handling_functions import CutoffTimeBasedStragglerHandling
+from openfl.component.straggler_handling_policy import CutoffTimeBasedStragglerHandling
 from openfl.databases import TensorDB
 from openfl.interface.aggregation_functions import WeightedAverage
 from openfl.pipelines import NoCompressionPipeline, TensorCodec
diff --git a/openfl/component/straggler_handling_functions/__init__.py b/openfl/component/straggler_handling_functions/__init__.py
deleted file mode 100644
index 5ab0af1794..0000000000
--- a/openfl/component/straggler_handling_functions/__init__.py
+++ /dev/null
@@ -1,13 +0,0 @@
-# Copyright 2020-2024 Intel Corporation
-# SPDX-License-Identifier: Apache-2.0
-
-
-from openfl.component.straggler_handling_functions.cutoff_time_based_straggler_handling import (
-    CutoffTimeBasedStragglerHandling,
-)
-from openfl.component.straggler_handling_functions.percentage_based_straggler_handling import (
-    PercentageBasedStragglerHandling,
-)
-from openfl.component.straggler_handling_functions.straggler_handling_function import (
-    StragglerHandlingPolicy,
-)
diff --git a/openfl/component/straggler_handling_functions/percentage_based_straggler_handling.py b/openfl/component/straggler_handling_functions/percentage_based_straggler_handling.py
deleted file mode 100644
index e556ea291b..0000000000
--- a/openfl/component/straggler_handling_functions/percentage_based_straggler_handling.py
+++ /dev/null
@@ -1,78 +0,0 @@
-# Copyright 2020-2024 Intel Corporation
-# SPDX-License-Identifier: Apache-2.0
-
-
-"""Percentage based Straggler Handling function."""
-from logging import getLogger
-
-from openfl.component.straggler_handling_functions.straggler_handling_function import (
-    StragglerHandlingPolicy,
-)
-
-
-class PercentageBasedStragglerHandling(StragglerHandlingPolicy):
-    """Percentage based Straggler Handling function."""
-
-    def __init__(self, percent_collaborators_needed=1.0, minimum_reporting=1, **kwargs):
-        """Initialize a PercentageBasedStragglerHandling object.
-
-        Args:
-            percent_collaborators_needed (float, optional): The percentage of
-                collaborators needed. Defaults to 1.0.
-            minimum_reporting (int, optional): The minimum number of
-                collaborators that should report. Defaults to 1.
-            **kwargs: Variable length argument list.
-        """
-        if minimum_reporting <= 0:
-            raise ValueError("minimum_reporting must be >0")
-
-        self.percent_collaborators_needed = percent_collaborators_needed
-        self.minimum_reporting = minimum_reporting
-        self.logger = getLogger(__name__)
-
-    def reset_policy_for_round(self) -> None:
-        """
-        Not required in PercentageBasedStragglerHandling.
-        """
-        pass
-
-    def start_policy(self, **kwargs) -> None:
-        """
-        Not required in PercentageBasedStragglerHandling.
-        """
-        pass
-
-    def straggler_cutoff_check(
-        self,
-        num_collaborators_done: int,
-        num_all_collaborators: int,
-    ) -> bool:
-        """
-        If percent_collaborators_needed and minimum_reporting collaborators have
-        reported results, then it is time to end round early.
-
-        Args:
-            num_collaborators_done (int): The number of collaborators that
-                have reported.
-            all_collaborators (list): All the collaborators.
-
-        Returns:
-            bool: True if the straggler cutoff conditions are met, False
-                otherwise.
-        """
-        return (
-            num_collaborators_done >= self.percent_collaborators_needed * num_all_collaborators
-        ) and self.__minimum_collaborators_reported(num_collaborators_done)
-
-    def __minimum_collaborators_reported(self, num_collaborators_done) -> bool:
-        """Check if the minimum number of collaborators have reported.
-
-        Args:
-            num_collaborators_done (int): The number of collaborators that
-                have reported.
-
-        Returns:
-            bool: True if the minimum number of collaborators have reported,
-                False otherwise.
-        """
-        return num_collaborators_done >= self.minimum_reporting
diff --git a/openfl/component/straggler_handling_functions/straggler_handling_function.py b/openfl/component/straggler_handling_functions/straggler_handling_function.py
deleted file mode 100644
index 8bd47bc045..0000000000
--- a/openfl/component/straggler_handling_functions/straggler_handling_function.py
+++ /dev/null
@@ -1,59 +0,0 @@
-# Copyright 2020-2024 Intel Corporation
-# SPDX-License-Identifier: Apache-2.0
-
-
-"""Straggler handling module."""
-
-from abc import ABC, abstractmethod
-
-
-class StragglerHandlingPolicy(ABC):
-    """Federated Learning straggler handling interface."""
-
-    @abstractmethod
-    def start_policy(self, **kwargs) -> None:
-        """
-        Start straggler handling policy for collaborator for a particular round.
-        NOTE: Refer CutoffTimeBasedStragglerHandling for reference.
-
-        Args:
-            **kwargs
-
-        Returns:
-            None
-        """
-        raise NotImplementedError
-
-    @abstractmethod
-    def reset_policy_for_round(self) -> None:
-        """
-        Reset policy variable for the next round.
-
-        Args:
-            None
-
-        Returns:
-            None
-        """
-        raise NotImplementedError
-
-    @abstractmethod
-    def straggler_cutoff_check(
-        self, num_collaborators_done: int, num_all_collaborators: int, **kwargs
-    ) -> bool:
-        """
-        Determines whether it is time to end the round early.
-
-        Args:
-            num_collaborators_done: int
-                Number of collaborators finished.
-            num_all_collaborators: int
-                Total number of collaborators.
-
-        Returns:
-            bool: True if it is time to end the round early, False otherwise.
-
-        Raises:
-            NotImplementedError: This method must be implemented by a subclass.
-        """
-        raise NotImplementedError
diff --git a/openfl/component/straggler_handling_policy/__init__.py b/openfl/component/straggler_handling_policy/__init__.py
new file mode 100644
index 0000000000..4de28d0e48
--- /dev/null
+++ b/openfl/component/straggler_handling_policy/__init__.py
@@ -0,0 +1,9 @@
+# Copyright 2020-2024 Intel Corporation
+# SPDX-License-Identifier: Apache-2.0
+
+
+from openfl.component.straggler_handling_policy.straggler_handling_policy import (
+    CutoffTimeBasedStragglerHandling,
+    PercentageBasedStragglerHandling,
+    StragglerHandlingPolicy,
+)
diff --git a/openfl/component/straggler_handling_functions/cutoff_time_based_straggler_handling.py b/openfl/component/straggler_handling_policy/straggler_handling_policy.py
similarity index 54%
rename from openfl/component/straggler_handling_functions/cutoff_time_based_straggler_handling.py
rename to openfl/component/straggler_handling_policy/straggler_handling_policy.py
index ca8e218f7c..3056b5d7b8 100644
--- a/openfl/component/straggler_handling_functions/cutoff_time_based_straggler_handling.py
+++ b/openfl/component/straggler_handling_policy/straggler_handling_policy.py
@@ -2,17 +2,67 @@
 # SPDX-License-Identifier: Apache-2.0
 
 
-"""Cutoff time based Straggler Handling function."""
+"""Straggler handling module."""
+
 import threading
 import time
+from abc import ABC, abstractmethod
 from logging import getLogger
 from typing import Callable
 
 import numpy as np
 
-from openfl.component.straggler_handling_functions.straggler_handling_function import (
-    StragglerHandlingPolicy,
-)
+
+class StragglerHandlingPolicy(ABC):
+    """Federated Learning straggler handling interface."""
+
+    @abstractmethod
+    def start_policy(self, **kwargs) -> None:
+        """
+        Start straggler handling policy for collaborator for a particular round.
+        NOTE: Refer CutoffTimeBasedStragglerHandling for reference.
+
+        Args:
+            **kwargs
+
+        Returns:
+            None
+        """
+        raise NotImplementedError
+
+    @abstractmethod
+    def reset_policy_for_round(self) -> None:
+        """
+        Reset policy variable for the next round.
+
+        Args:
+            None
+
+        Returns:
+            None
+        """
+        raise NotImplementedError
+
+    @abstractmethod
+    def straggler_cutoff_check(
+        self, num_collaborators_done: int, num_all_collaborators: int, **kwargs
+    ) -> bool:
+        """
+        Determines whether it is time to end the round early.
+
+        Args:
+            num_collaborators_done: int
+                Number of collaborators finished.
+            num_all_collaborators: int
+                Total number of collaborators.
+
+        Returns:
+            bool: True if it is time to end the round early, False otherwise.
+
+        Raises:
+            NotImplementedError: This method must be implemented by a subclass.
+        """
+        raise NotImplementedError
 
 
 class CutoffTimeBasedStragglerHandling(StragglerHandlingPolicy):
@@ -140,3 +190,71 @@ def __minimum_collaborators_reported(self, num_collaborators_done) -> bool:
                 False otherwise.
         """
         return num_collaborators_done >= self.minimum_reporting
+
+
+class PercentageBasedStragglerHandling(StragglerHandlingPolicy):
+    """Percentage based Straggler Handling function."""
+
+    def __init__(self, percent_collaborators_needed=1.0, minimum_reporting=1, **kwargs):
+        """Initialize a PercentageBasedStragglerHandling object.
+
+        Args:
+            percent_collaborators_needed (float, optional): The percentage of
+                collaborators needed. Defaults to 1.0.
+            minimum_reporting (int, optional): The minimum number of
+                collaborators that should report. Defaults to 1.
+            **kwargs: Variable length argument list.
+        """
+        if minimum_reporting <= 0:
+            raise ValueError("minimum_reporting must be >0")
+
+        self.percent_collaborators_needed = percent_collaborators_needed
+        self.minimum_reporting = minimum_reporting
+        self.logger = getLogger(__name__)
+
+    def reset_policy_for_round(self) -> None:
+        """
+        Not required in PercentageBasedStragglerHandling.
+        """
+        pass
+
+    def start_policy(self, **kwargs) -> None:
+        """
+        Not required in PercentageBasedStragglerHandling.
+        """
+        pass
+
+    def straggler_cutoff_check(
+        self,
+        num_collaborators_done: int,
+        num_all_collaborators: int,
+    ) -> bool:
+        """
+        If percent_collaborators_needed and minimum_reporting collaborators have
+        reported results, then it is time to end round early.
+
+        Args:
+            num_collaborators_done (int): The number of collaborators that
+                have reported.
+            all_collaborators (list): All the collaborators.
+
+        Returns:
+            bool: True if the straggler cutoff conditions are met, False
+                otherwise.
+        """
+        return (
+            num_collaborators_done >= self.percent_collaborators_needed * num_all_collaborators
+        ) and self.__minimum_collaborators_reported(num_collaborators_done)
+
+    def __minimum_collaborators_reported(self, num_collaborators_done) -> bool:
+        """Check if the minimum number of collaborators have reported.
+
+        Args:
+            num_collaborators_done (int): The number of collaborators that
+                have reported.
+
+        Returns:
+            bool: True if the minimum number of collaborators have reported,
+                False otherwise.
+        """
+        return num_collaborators_done >= self.minimum_reporting
diff --git a/openfl/federated/plan/plan.py b/openfl/federated/plan/plan.py
index 5f0575837d..526e0d49c2 100644
--- a/openfl/federated/plan/plan.py
+++ b/openfl/federated/plan/plan.py
@@ -421,7 +421,7 @@ def get_tensor_pipe(self):
 
     def get_straggler_handling_policy(self):
         """Get straggler handling policy."""
-        template = "openfl.component.straggler_handling_functions.CutoffTimeBasedStragglerHandling"
+        template = "openfl.component.straggler_handling_policy.CutoffTimeBasedStragglerHandling"
         defaults = self.config.get("straggler_handling_policy", {TEMPLATE: template, SETTINGS: {}})
 
         if self.straggler_policy_ is None:
diff --git a/setup.py b/setup.py
index b1dea5bcfd..27fc79ac98 100644
--- a/setup.py
+++ b/setup.py
@@ -97,7 +97,7 @@ def run(self):
         'openfl.component.collaborator',
         'openfl.component.director',
         'openfl.component.envoy',
-        'openfl.component.straggler_handling_functions',
+        'openfl.component.straggler_handling_policy',
         'openfl.cryptography',
         'openfl.databases',
         'openfl.databases.utilities',

From 896786f97a437ffa9e3f407b4acee85fbac00cda Mon Sep 17 00:00:00 2001
From: Ishant Thakare <ishantrog752@gmail.com>
Date: Wed, 23 Oct 2024 17:55:43 +0530
Subject: [PATCH 2/6] Incorporated review comments

Signed-off-by: Ishant Thakare <ishantrog752@gmail.com>
---
 .../straggler_handling_algorithms.rst         |  8 +++----
 docs/source/api/openfl_component.rst          |  1 -
 .../plan/plan.yaml                            |  2 +-
 openfl/component/__init__.py                  | 10 ++++----
 openfl/component/aggregator/__init__.py       |  5 ++++
 openfl/component/aggregator/aggregator.py     | 14 ++++++-----
 .../straggler_handling.py}                    | 24 ++++++++++---------
 .../straggler_handling_policy/__init__.py     |  9 -------
 openfl/federated/plan/plan.py                 |  2 +-
 setup.py                                      |  1 -
 10 files changed, 37 insertions(+), 39 deletions(-)
 rename openfl/component/{straggler_handling_policy/straggler_handling_policy.py => aggregator/straggler_handling.py} (92%)
 delete mode 100644 openfl/component/straggler_handling_policy/__init__.py

diff --git a/docs/developer_guide/advanced_topics/straggler_handling_algorithms.rst b/docs/developer_guide/advanced_topics/straggler_handling_algorithms.rst
index d5f00500e0..26527de921 100644
--- a/docs/developer_guide/advanced_topics/straggler_handling_algorithms.rst
+++ b/docs/developer_guide/advanced_topics/straggler_handling_algorithms.rst
@@ -11,7 +11,7 @@ The Open Federated Learning (|productName|) framework supports straggler handlin
 
 The following are the straggler handling algorithms supported in |productName|:
 
-``CutoffTimeBasedStragglerHandling``
+``CutoffPolicy``
     Identifies stragglers based on the cutoff time specified in the settings. Arguments to the function are:
         - *Cutoff Time* (straggler_cutoff_time), specifies the cutoff time by which the aggregator should end the round early.
         - *Minimum Reporting* (minimum_reporting), specifies the minimum number of collaborators needed to aggregate the model.
@@ -19,7 +19,7 @@ The following are the straggler handling algorithms supported in |productName|:
     For example, in a federation of 5 collaborators, if :code:`straggler_cutoff_time` (in seconds) is set to 20 and :code:`minimum_reporting` is set to 2, atleast 2 collaborators (or more) would be included in the round, provided that the time limit of 20 seconds is not exceeded.
     In an event where :code:`minimum_reporting` collaborators don't make it within the :code:`straggler_cutoff_time`, the straggler handling policy is disregarded. 
 
-``PercentageBasedStragglerHandling``
+``PercentagePolicy``
     Identifies stragglers based on the percetage specified. Arguments to the function are:
         - *Percentage of collaborators* (percent_collaborators_needed), specifies a percentage of collaborators enough to end the round early.
         - *Minimum Reporting* (minimum_reporting), specifies the minimum number of collaborators needed to aggregate the model.
@@ -29,12 +29,12 @@ The following are the straggler handling algorithms supported in |productName|:
 Demonstration of adding the straggler handling interface
 =========================================================
 
-The example template, **torch_cnn_mnist_straggler_check**, uses the ``PercentageBasedStragglerHandling``. To gain a better understanding of how experiments perform, you can modify the **percent_collaborators_needed** or **minimum_reporting** parameter in the template **plan.yaml** or even choose **CutoffTimeBasedStragglerHandling** function instead:
+The example template, **torch_cnn_mnist_straggler_check**, uses the ``PercentagePolicy``. To gain a better understanding of how experiments perform, you can modify the **percent_collaborators_needed** or **minimum_reporting** parameter in the template **plan.yaml** or even choose **CutoffPolicy** function instead:
 
     .. code-block:: yaml
     
         straggler_handling_policy :
-            template : openfl.component.straggler_handling_policy.CutoffTimeBasedStragglerHandling
+            template : openfl.component.aggregator.straggler_handling.CutoffPolicy
             settings :
                 straggler_cutoff_time : 20
                 minimum_reporting : 1
diff --git a/docs/source/api/openfl_component.rst b/docs/source/api/openfl_component.rst
index 67639599f5..0af3099b49 100644
--- a/docs/source/api/openfl_component.rst
+++ b/docs/source/api/openfl_component.rst
@@ -17,4 +17,3 @@ Component modules reference:
    openfl.component.collaborator
    openfl.component.director
    openfl.component.envoy
-   openfl.component.straggler_handling_policy
diff --git a/openfl-workspace/torch_cnn_mnist_straggler_check/plan/plan.yaml b/openfl-workspace/torch_cnn_mnist_straggler_check/plan/plan.yaml
index cca9414fb6..e2378424b9 100644
--- a/openfl-workspace/torch_cnn_mnist_straggler_check/plan/plan.yaml
+++ b/openfl-workspace/torch_cnn_mnist_straggler_check/plan/plan.yaml
@@ -45,7 +45,7 @@ compression_pipeline :
   defaults : plan/defaults/compression_pipeline.yaml
 
 straggler_handling_policy :
-  template : openfl.component.straggler_handling_policy.CutoffTimeBasedStragglerHandling
+  template : openfl.component.aggregator.straggler_handling.PercentagePolicy
   settings : 
     percent_collaborators_needed : 0.5
     minimum_reporting : 1
\ No newline at end of file
diff --git a/openfl/component/__init__.py b/openfl/component/__init__.py
index 97788f4a6f..60a078371e 100644
--- a/openfl/component/__init__.py
+++ b/openfl/component/__init__.py
@@ -3,12 +3,12 @@
 
 
 from openfl.component.aggregator.aggregator import Aggregator
+from openfl.component.aggregator.straggler_handling import (
+    CutoffPolicy,
+    PercentagePolicy,
+    StragglerHandlingPolicy,
+)
 from openfl.component.assigner.assigner import Assigner
 from openfl.component.assigner.random_grouped_assigner import RandomGroupedAssigner
 from openfl.component.assigner.static_grouped_assigner import StaticGroupedAssigner
 from openfl.component.collaborator.collaborator import Collaborator
-from openfl.component.straggler_handling_policy import (
-    CutoffTimeBasedStragglerHandling,
-    PercentageBasedStragglerHandling,
-    StragglerHandlingPolicy,
-)
diff --git a/openfl/component/aggregator/__init__.py b/openfl/component/aggregator/__init__.py
index ed7661486e..728e4c68d8 100644
--- a/openfl/component/aggregator/__init__.py
+++ b/openfl/component/aggregator/__init__.py
@@ -3,3 +3,8 @@
 
 
 from openfl.component.aggregator.aggregator import Aggregator
+from openfl.component.aggregator.straggler_handling import (
+    CutoffPolicy,
+    PercentagePolicy,
+    StragglerHandlingPolicy,
+)
diff --git a/openfl/component/aggregator/aggregator.py b/openfl/component/aggregator/aggregator.py
index 4de7ab40af..eb23bc50ed 100644
--- a/openfl/component/aggregator/aggregator.py
+++ b/openfl/component/aggregator/aggregator.py
@@ -8,7 +8,7 @@
 from logging import getLogger
 from threading import Lock
 
-from openfl.component.straggler_handling_policy import CutoffTimeBasedStragglerHandling
+from openfl.component.aggregator.straggler_handling import CutoffPolicy
 from openfl.databases import TensorDB
 from openfl.interface.aggregation_functions import WeightedAverage
 from openfl.pipelines import NoCompressionPipeline, TensorCodec
@@ -69,7 +69,7 @@ def __init__(
         best_state_path,
         last_state_path,
         assigner,
-        straggler_handling_policy=None,
+        straggler_handling_policy=CutoffPolicy,
         rounds_to_train=256,
         single_col_cert_common_name=None,
         compression_pipeline=None,
@@ -92,7 +92,7 @@ def __init__(
                 weight.
             assigner: Assigner object.
             straggler_handling_policy (optional): Straggler handling policy.
-                Defaults to CutoffTimeBasedStragglerHandling.
+                Defaults to CutoffPolicy.
             rounds_to_train (int, optional): Number of rounds to train.
                 Defaults to 256.
             single_col_cert_common_name (str, optional): Common name for single
@@ -117,9 +117,11 @@ def __init__(
             # Cleaner solution?
             self.single_col_cert_common_name = ""
 
-        self.straggler_handling_policy = (
-            straggler_handling_policy or CutoffTimeBasedStragglerHandling()
-        )
+        if straggler_handling_policy == CutoffPolicy:
+            self.straggler_handling_policy = straggler_handling_policy()
+        else:
+            self.straggler_handling_policy = straggler_handling_policy
+
         self._end_of_round_check_done = [False] * rounds_to_train
         self.stragglers = []
 
diff --git a/openfl/component/straggler_handling_policy/straggler_handling_policy.py b/openfl/component/aggregator/straggler_handling.py
similarity index 92%
rename from openfl/component/straggler_handling_policy/straggler_handling_policy.py
rename to openfl/component/aggregator/straggler_handling.py
index 3056b5d7b8..df26080e90 100644
--- a/openfl/component/straggler_handling_policy/straggler_handling_policy.py
+++ b/openfl/component/aggregator/straggler_handling.py
@@ -20,7 +20,7 @@ class StragglerHandlingPolicy(ABC):
     def start_policy(self, **kwargs) -> None:
         """
         Start straggler handling policy for collaborator for a particular round.
-        NOTE: Refer CutoffTimeBasedStragglerHandling for reference.
+        NOTE: Refer CutoffPolicy for reference.
 
         Args:
             **kwargs
@@ -65,14 +65,14 @@ def straggler_cutoff_check(
         raise NotImplementedError
 
 
-class CutoffTimeBasedStragglerHandling(StragglerHandlingPolicy):
+class CutoffPolicy(StragglerHandlingPolicy):
     """Cutoff time based Straggler Handling function."""
 
     def __init__(
         self, round_start_time=None, straggler_cutoff_time=np.inf, minimum_reporting=1, **kwargs
     ):
         """
-         Initialize a CutoffTimeBasedStragglerHandling object.
+         Initialize a CutoffPolicy object.
 
         Args:
             round_start_time (optional): The start time of the round. Defaults
@@ -89,12 +89,12 @@ def __init__(
         self.round_start_time = round_start_time
         self.straggler_cutoff_time = straggler_cutoff_time
         self.minimum_reporting = minimum_reporting
+        self.is_timer_started = False
         self.logger = getLogger(__name__)
 
         if self.straggler_cutoff_time == np.inf:
             self.logger.warning(
-                "CutoffTimeBasedStragglerHandling is disabled as straggler_cutoff_time "
-                "is set to np.inf."
+                "CutoffPolicy is disabled as straggler_cutoff_time " "is set to np.inf."
             )
 
     def reset_policy_for_round(self) -> None:
@@ -103,7 +103,7 @@ def reset_policy_for_round(self) -> None:
         """
         if hasattr(self, "timer"):
             self.timer.cancel()
-            delattr(self, "timer")
+        self.is_timer_started = False
 
     def start_policy(self, callback: Callable) -> None:
         """
@@ -120,8 +120,9 @@ def start_policy(self, callback: Callable) -> None:
         # If straggler_cutoff_time is set to infinity
         # or if the timer is already running,
         # do not start the policy.
-        if self.straggler_cutoff_time == np.inf or hasattr(self, "timer"):
+        if self.straggler_cutoff_time == np.inf or self.is_timer_started:
             return
+
         self.round_start_time = time.time()
         self.timer = threading.Timer(
             self.straggler_cutoff_time,
@@ -129,6 +130,7 @@ def start_policy(self, callback: Callable) -> None:
         )
         self.timer.daemon = True
         self.timer.start()
+        self.is_timer_started = True
 
     def straggler_cutoff_check(
         self,
@@ -192,11 +194,11 @@ def __minimum_collaborators_reported(self, num_collaborators_done) -> bool:
         return num_collaborators_done >= self.minimum_reporting
 
 
-class PercentageBasedStragglerHandling(StragglerHandlingPolicy):
+class PercentagePolicy(StragglerHandlingPolicy):
     """Percentage based Straggler Handling function."""
 
     def __init__(self, percent_collaborators_needed=1.0, minimum_reporting=1, **kwargs):
-        """Initialize a PercentageBasedStragglerHandling object.
+        """Initialize a PercentagePolicy object.
 
         Args:
             percent_collaborators_needed (float, optional): The percentage of
@@ -214,13 +216,13 @@ def __init__(self, percent_collaborators_needed=1.0, minimum_reporting=1, **kwar
 
     def reset_policy_for_round(self) -> None:
         """
-        Not required in PercentageBasedStragglerHandling.
+        Not required in PercentagePolicy.
         """
         pass
 
     def start_policy(self, **kwargs) -> None:
         """
-        Not required in PercentageBasedStragglerHandling.
+        Not required in PercentagePolicy.
         """
         pass
 
diff --git a/openfl/component/straggler_handling_policy/__init__.py b/openfl/component/straggler_handling_policy/__init__.py
deleted file mode 100644
index 4de28d0e48..0000000000
--- a/openfl/component/straggler_handling_policy/__init__.py
+++ /dev/null
@@ -1,9 +0,0 @@
-# Copyright 2020-2024 Intel Corporation
-# SPDX-License-Identifier: Apache-2.0
-
-
-from openfl.component.straggler_handling_policy.straggler_handling_policy import (
-    CutoffTimeBasedStragglerHandling,
-    PercentageBasedStragglerHandling,
-    StragglerHandlingPolicy,
-)
diff --git a/openfl/federated/plan/plan.py b/openfl/federated/plan/plan.py
index 526e0d49c2..10ef43589f 100644
--- a/openfl/federated/plan/plan.py
+++ b/openfl/federated/plan/plan.py
@@ -421,7 +421,7 @@ def get_tensor_pipe(self):
 
     def get_straggler_handling_policy(self):
         """Get straggler handling policy."""
-        template = "openfl.component.straggler_handling_policy.CutoffTimeBasedStragglerHandling"
+        template = "openfl.component.aggregator.straggler_handling.CutoffPolicy"
         defaults = self.config.get("straggler_handling_policy", {TEMPLATE: template, SETTINGS: {}})
 
         if self.straggler_policy_ is None:
diff --git a/setup.py b/setup.py
index bb64069e07..c6b03dcfc1 100644
--- a/setup.py
+++ b/setup.py
@@ -97,7 +97,6 @@ def run(self):
         'openfl.component.collaborator',
         'openfl.component.director',
         'openfl.component.envoy',
-        'openfl.component.straggler_handling_policy',
         'openfl.cryptography',
         'openfl.databases',
         'openfl.databases.utilities',

From eff79fc52a5c5b59de958c510837127459380f91 Mon Sep 17 00:00:00 2001
From: Ishant Thakare <ishantrog752@gmail.com>
Date: Fri, 20 Dec 2024 14:38:11 +0530
Subject: [PATCH 3/6] Incorporated karan's review comments

Signed-off-by: Ishant Thakare <ishantrog752@gmail.com>
---
 openfl/component/__init__.py                  |  2 +-
 openfl/component/aggregator/__init__.py       |  2 +-
 openfl/component/aggregator/aggregator.py     | 10 ++--
 .../aggregator/straggler_handling.py          | 46 +++++--------------
 openfl/federated/plan/plan.py                 |  8 +++-
 5 files changed, 23 insertions(+), 45 deletions(-)

diff --git a/openfl/component/__init__.py b/openfl/component/__init__.py
index 60a078371e..792cfd855d 100644
--- a/openfl/component/__init__.py
+++ b/openfl/component/__init__.py
@@ -6,7 +6,7 @@
 from openfl.component.aggregator.straggler_handling import (
     CutoffPolicy,
     PercentagePolicy,
-    StragglerHandlingPolicy,
+    StragglerPolicy,
 )
 from openfl.component.assigner.assigner import Assigner
 from openfl.component.assigner.random_grouped_assigner import RandomGroupedAssigner
diff --git a/openfl/component/aggregator/__init__.py b/openfl/component/aggregator/__init__.py
index 728e4c68d8..1bbddaf4f2 100644
--- a/openfl/component/aggregator/__init__.py
+++ b/openfl/component/aggregator/__init__.py
@@ -6,5 +6,5 @@
 from openfl.component.aggregator.straggler_handling import (
     CutoffPolicy,
     PercentagePolicy,
-    StragglerHandlingPolicy,
+    StragglerPolicy,
 )
diff --git a/openfl/component/aggregator/aggregator.py b/openfl/component/aggregator/aggregator.py
index 354d512d15..db718dfc85 100644
--- a/openfl/component/aggregator/aggregator.py
+++ b/openfl/component/aggregator/aggregator.py
@@ -9,7 +9,7 @@
 from logging import getLogger
 from threading import Lock
 
-from openfl.component.aggregator.straggler_handling import CutoffPolicy
+from openfl.component.aggregator.straggler_handling import CutoffPolicy, StragglerPolicy
 from openfl.databases import TensorDB
 from openfl.interface.aggregation_functions import WeightedAverage
 from openfl.pipelines import NoCompressionPipeline, TensorCodec
@@ -71,7 +71,7 @@ def __init__(
         last_state_path,
         assigner,
         use_delta_updates=True,
-        straggler_handling_policy=CutoffPolicy,
+        straggler_handling_policy: StragglerPolicy = CutoffPolicy,
         rounds_to_train=256,
         single_col_cert_common_name=None,
         compression_pipeline=None,
@@ -95,7 +95,6 @@ def __init__(
                 weight.
             assigner: Assigner object.
             straggler_handling_policy (optional): Straggler handling policy.
-                Defaults to CutoffPolicy.
             rounds_to_train (int, optional): Number of rounds to train.
                 Defaults to 256.
             single_col_cert_common_name (str, optional): Common name for single
@@ -123,10 +122,7 @@ def __init__(
         # FIXME: "" instead of None is for protobuf compatibility.
         self.single_col_cert_common_name = single_col_cert_common_name or ""
 
-        if straggler_handling_policy == CutoffPolicy:
-            self.straggler_handling_policy = straggler_handling_policy()
-        else:
-            self.straggler_handling_policy = straggler_handling_policy
+        self.straggler_handling_policy = straggler_handling_policy()
 
         self._end_of_round_check_done = [False] * rounds_to_train
         self.stragglers = []
diff --git a/openfl/component/aggregator/straggler_handling.py b/openfl/component/aggregator/straggler_handling.py
index df26080e90..1d5f6ef7de 100644
--- a/openfl/component/aggregator/straggler_handling.py
+++ b/openfl/component/aggregator/straggler_handling.py
@@ -12,8 +12,10 @@
 
 import numpy as np
 
+logger = getLogger(__name__)
 
-class StragglerHandlingPolicy(ABC):
+
+class StragglerPolicy(ABC):
     """Federated Learning straggler handling interface."""
 
     @abstractmethod
@@ -24,23 +26,12 @@ def start_policy(self, **kwargs) -> None:
 
         Args:
             **kwargs
-
-        Returns:
-            None
         """
         raise NotImplementedError
 
     @abstractmethod
     def reset_policy_for_round(self) -> None:
-        """
-        Reset policy variable for the next round.
-
-        Args:
-            None
-
-        Returns:
-            None
-        """
+        """Reset policy for the next round."""
         raise NotImplementedError
 
     @abstractmethod
@@ -65,7 +56,7 @@ def straggler_cutoff_check(
         raise NotImplementedError
 
 
-class CutoffPolicy(StragglerHandlingPolicy):
+class CutoffPolicy(StragglerPolicy):
     """Cutoff time based Straggler Handling function."""
 
     def __init__(
@@ -90,17 +81,12 @@ def __init__(
         self.straggler_cutoff_time = straggler_cutoff_time
         self.minimum_reporting = minimum_reporting
         self.is_timer_started = False
-        self.logger = getLogger(__name__)
 
         if self.straggler_cutoff_time == np.inf:
-            self.logger.warning(
-                "CutoffPolicy is disabled as straggler_cutoff_time " "is set to np.inf."
-            )
+            logger.warning("CutoffPolicy is disabled as straggler_cutoff_time " "is set to np.inf.")
 
     def reset_policy_for_round(self) -> None:
-        """
-        Reset timer for the next round.
-        """
+        """Reset timer for the next round."""
         if hasattr(self, "timer"):
             self.timer.cancel()
         self.is_timer_started = False
@@ -113,9 +99,6 @@ def start_policy(self, callback: Callable) -> None:
         Args:
             callback: Callable
                 Callback function for when straggler_cutoff_time elapses
-
-        Returns:
-            None
         """
         # If straggler_cutoff_time is set to infinity
         # or if the timer is already running,
@@ -159,13 +142,13 @@ def straggler_cutoff_check(
         # Time has expired
         # Check if minimum_reporting collaborators have reported results
         elif self.__minimum_collaborators_reported(num_collaborators_done):
-            self.logger.info(
+            logger.info(
                 f"{num_collaborators_done} collaborators have reported results. "
                 "Applying cutoff policy and proceeding with end of round."
             )
             return True
         else:
-            self.logger.info(
+            logger.info(
                 f"Waiting for minimum {self.minimum_reporting} collaborator(s) to report results."
             )
             return False
@@ -194,7 +177,7 @@ def __minimum_collaborators_reported(self, num_collaborators_done) -> bool:
         return num_collaborators_done >= self.minimum_reporting
 
 
-class PercentagePolicy(StragglerHandlingPolicy):
+class PercentagePolicy(StragglerPolicy):
     """Percentage based Straggler Handling function."""
 
     def __init__(self, percent_collaborators_needed=1.0, minimum_reporting=1, **kwargs):
@@ -212,18 +195,13 @@ def __init__(self, percent_collaborators_needed=1.0, minimum_reporting=1, **kwar
 
         self.percent_collaborators_needed = percent_collaborators_needed
         self.minimum_reporting = minimum_reporting
-        self.logger = getLogger(__name__)
 
     def reset_policy_for_round(self) -> None:
-        """
-        Not required in PercentagePolicy.
-        """
+        """Not required in PercentagePolicy."""
         pass
 
     def start_policy(self, **kwargs) -> None:
-        """
-        Not required in PercentagePolicy.
-        """
+        """Not required in PercentagePolicy."""
         pass
 
     def straggler_cutoff_check(
diff --git a/openfl/federated/plan/plan.py b/openfl/federated/plan/plan.py
index 9c326a746a..b2321d2011 100644
--- a/openfl/federated/plan/plan.py
+++ b/openfl/federated/plan/plan.py
@@ -4,6 +4,7 @@
 
 """Plan module."""
 
+from functools import partial
 from hashlib import sha384
 from importlib import import_module
 from logging import getLogger
@@ -43,7 +44,7 @@ class Plan:
         server_ (AggregatorGRPCServer): gRPC server object.
         client_ (AggregatorGRPCClient): gRPC client object.
         pipe_ (CompressionPipeline): Compression pipeline object.
-        straggler_policy_ (StragglerHandlingPolicy): Straggler handling policy.
+        straggler_policy_ (StragglerPolicy): Straggler handling policy.
         hash_ (str): Hash of the instance.
         name_ (str): Name of the instance.
         serializer_ (SerializerPlugin): Serializer plugin.
@@ -426,7 +427,10 @@ def get_straggler_handling_policy(self):
         defaults = self.config.get("straggler_handling_policy", {TEMPLATE: template, SETTINGS: {}})
 
         if self.straggler_policy_ is None:
-            self.straggler_policy_ = Plan.build(**defaults)
+            # Prepare a partial function for the straggler policy
+            self.straggler_policy_ = partial(
+                Plan.import_(defaults["template"]), **defaults["settings"]
+            )
 
         return self.straggler_policy_
 

From 7f16cdb77bd3a7fd1b9b38daf8791fde9f316ebf Mon Sep 17 00:00:00 2001
From: Ishant Thakare <ishantrog752@gmail.com>
Date: Fri, 20 Dec 2024 14:55:41 +0530
Subject: [PATCH 4/6] Resolving merge conflicts

Signed-off-by: Ishant Thakare <ishantrog752@gmail.com>
---
 docs/source/api/openfl_component.rst     | 19 -------------------
 docs/source/api/openfl_cryptography.rst  | 16 ----------------
 docs/source/api/openfl_databases.rst     | 16 ----------------
 docs/source/api/openfl_experimental.rst  | 18 ------------------
 docs/source/api/openfl_federated.rst     | 18 ------------------
 docs/source/api/openfl_interface.rst     | 16 ----------------
 docs/source/api/openfl_native.rst        | 16 ----------------
 docs/source/api/openfl_pipelines.rst     | 16 ----------------
 docs/source/api/openfl_plugins.rst       | 16 ----------------
 docs/source/api/openfl_protocols.rst     | 16 ----------------
 docs/source/api/openfl_transport.rst     | 15 ---------------
 docs/source/api/openfl_utilities.rst     | 16 ----------------
 docs/source/openfl/director_workflow.svg |  1 -
 docs/source/openfl/static_diagram.svg    |  1 -
 14 files changed, 200 deletions(-)
 delete mode 100644 docs/source/api/openfl_component.rst
 delete mode 100644 docs/source/api/openfl_cryptography.rst
 delete mode 100644 docs/source/api/openfl_databases.rst
 delete mode 100644 docs/source/api/openfl_experimental.rst
 delete mode 100644 docs/source/api/openfl_federated.rst
 delete mode 100644 docs/source/api/openfl_interface.rst
 delete mode 100644 docs/source/api/openfl_native.rst
 delete mode 100644 docs/source/api/openfl_pipelines.rst
 delete mode 100644 docs/source/api/openfl_plugins.rst
 delete mode 100644 docs/source/api/openfl_protocols.rst
 delete mode 100644 docs/source/api/openfl_transport.rst
 delete mode 100644 docs/source/api/openfl_utilities.rst
 delete mode 100644 docs/source/openfl/director_workflow.svg
 delete mode 100644 docs/source/openfl/static_diagram.svg

diff --git a/docs/source/api/openfl_component.rst b/docs/source/api/openfl_component.rst
deleted file mode 100644
index 0af3099b49..0000000000
--- a/docs/source/api/openfl_component.rst
+++ /dev/null
@@ -1,19 +0,0 @@
-.. # Copyright (C) 2020-2024 Intel Corporation
-.. # SPDX-License-Identifier: Apache-2.0
-
-*************************************************
-Component Module
-*************************************************
-
-Component modules reference:
-
-.. autosummary::
-   :toctree: _autosummary
-   :template: custom-module-template.rst
-   :recursive:
-
-   openfl.component.aggregator
-   openfl.component.assigner
-   openfl.component.collaborator
-   openfl.component.director
-   openfl.component.envoy
diff --git a/docs/source/api/openfl_cryptography.rst b/docs/source/api/openfl_cryptography.rst
deleted file mode 100644
index 475ebd1e9b..0000000000
--- a/docs/source/api/openfl_cryptography.rst
+++ /dev/null
@@ -1,16 +0,0 @@
-.. # Copyright (C) 2020-2024 Intel Corporation
-.. # SPDX-License-Identifier: Apache-2.0
-
-*************************************************
-Cryptography Module
-*************************************************
-
-Cryptography modules reference:
-
-.. autosummary::
-   :toctree: _autosummary
-   :recursive:
-
-   openfl.cryptography.ca
-   openfl.cryptography.io
-   openfl.cryptography.participant
diff --git a/docs/source/api/openfl_databases.rst b/docs/source/api/openfl_databases.rst
deleted file mode 100644
index 8014d42114..0000000000
--- a/docs/source/api/openfl_databases.rst
+++ /dev/null
@@ -1,16 +0,0 @@
-.. # Copyright (C) 2020-2024 Intel Corporation
-.. # SPDX-License-Identifier: Apache-2.0
-
-*************************************************
-Databases Module
-*************************************************
-
-Databases modules reference:
-
-.. autosummary::
-   :toctree: _autosummary
-   :template: custom-module-template.rst
-   :recursive:
-
-   openfl.databases
-   
\ No newline at end of file
diff --git a/docs/source/api/openfl_experimental.rst b/docs/source/api/openfl_experimental.rst
deleted file mode 100644
index 907645686d..0000000000
--- a/docs/source/api/openfl_experimental.rst
+++ /dev/null
@@ -1,18 +0,0 @@
-.. # Copyright (C) 2020-2024 Intel Corporation
-.. # SPDX-License-Identifier: Apache-2.0
-
-*************************************************
-Experimental Module
-*************************************************
-
-Experimental modules reference:
-
-.. autosummary::
-   :toctree: _autosummary
-   :template: custom-module-template.rst
-   :recursive:
-
-   openfl.experimental.workflow.interface
-   openfl.experimental.workflow.placement
-   openfl.experimental.workflow.runtime
-   openfl.experimental.workflow.utilities
diff --git a/docs/source/api/openfl_federated.rst b/docs/source/api/openfl_federated.rst
deleted file mode 100644
index 8c3d50b81e..0000000000
--- a/docs/source/api/openfl_federated.rst
+++ /dev/null
@@ -1,18 +0,0 @@
-.. # Copyright (C) 2020-2024 Intel Corporation
-.. # SPDX-License-Identifier: Apache-2.0
-
-*************************************************
-Federated Module
-*************************************************
-
-Federated modules reference:
-
-.. autosummary::
-   :toctree: _autosummary
-   :template: custom-module-template.rst
-   :recursive:
-
-   openfl.federated.plan
-   openfl.federated.task
-   openfl.federated.data
-   
\ No newline at end of file
diff --git a/docs/source/api/openfl_interface.rst b/docs/source/api/openfl_interface.rst
deleted file mode 100644
index 8685cce5f0..0000000000
--- a/docs/source/api/openfl_interface.rst
+++ /dev/null
@@ -1,16 +0,0 @@
-.. # Copyright (C) 2020-2024 Intel Corporation
-.. # SPDX-License-Identifier: Apache-2.0
-
-*************************************************
-Interface Module
-*************************************************
-
-Interface modules reference:
-
-.. autosummary::
-   :toctree: _autosummary
-   :template: custom-module-template.rst
-   :recursive:
-
-   openfl.interface
-   
\ No newline at end of file
diff --git a/docs/source/api/openfl_native.rst b/docs/source/api/openfl_native.rst
deleted file mode 100644
index 5f3f513340..0000000000
--- a/docs/source/api/openfl_native.rst
+++ /dev/null
@@ -1,16 +0,0 @@
-.. # Copyright (C) 2020-2024 Intel Corporation
-.. # SPDX-License-Identifier: Apache-2.0
-
-*************************************************
-Native Module (Deprecated)
-*************************************************
-
-Native modules reference:
-
-.. autosummary::
-   :toctree: _autosummary
-   :template: custom-module-template.rst
-   :recursive:
-
-   openfl.native
-   
\ No newline at end of file
diff --git a/docs/source/api/openfl_pipelines.rst b/docs/source/api/openfl_pipelines.rst
deleted file mode 100644
index 42ec1b33ad..0000000000
--- a/docs/source/api/openfl_pipelines.rst
+++ /dev/null
@@ -1,16 +0,0 @@
-.. # Copyright (C) 2020-2024 Intel Corporation
-.. # SPDX-License-Identifier: Apache-2.0
-
-*************************************************
-Pipelines Module
-*************************************************
-
-Pipelines modules reference:
-
-.. autosummary::
-   :toctree: _autosummary
-   :template: custom-module-template.rst
-   :recursive:
-
-   openfl.pipelines
-   
\ No newline at end of file
diff --git a/docs/source/api/openfl_plugins.rst b/docs/source/api/openfl_plugins.rst
deleted file mode 100644
index de8df91f4f..0000000000
--- a/docs/source/api/openfl_plugins.rst
+++ /dev/null
@@ -1,16 +0,0 @@
-.. # Copyright (C) 2020-2024 Intel Corporation
-.. # SPDX-License-Identifier: Apache-2.0
-
-*************************************************
-Plugins Module
-*************************************************
-
-Plugins modules reference:
-
-.. autosummary::
-   :toctree: _autosummary
-   :template: custom-module-template.rst
-   :recursive:
-
-   openfl.plugins
-   
\ No newline at end of file
diff --git a/docs/source/api/openfl_protocols.rst b/docs/source/api/openfl_protocols.rst
deleted file mode 100644
index e6e571ccc3..0000000000
--- a/docs/source/api/openfl_protocols.rst
+++ /dev/null
@@ -1,16 +0,0 @@
-.. # Copyright (C) 2020-2024 Intel Corporation
-.. # SPDX-License-Identifier: Apache-2.0
-
-*************************************************
-Protocols Module
-*************************************************
-
-Protocols modules reference:
-
-.. autosummary::
-   :toctree: _autosummary
-   :template: custom-module-template.rst
-   :recursive:
-
-   openfl.protocols
-   
\ No newline at end of file
diff --git a/docs/source/api/openfl_transport.rst b/docs/source/api/openfl_transport.rst
deleted file mode 100644
index 19eb01d839..0000000000
--- a/docs/source/api/openfl_transport.rst
+++ /dev/null
@@ -1,15 +0,0 @@
-.. # Copyright (C) 2020-2024 Intel Corporation
-.. # SPDX-License-Identifier: Apache-2.0
-
-*************************************************
-Transport Module
-*************************************************
-
-Transport modules reference:
-
-.. autosummary::
-   :toctree: _autosummary
-   :template: custom-module-template.rst
-   :recursive:
-
-   openfl.transport
diff --git a/docs/source/api/openfl_utilities.rst b/docs/source/api/openfl_utilities.rst
deleted file mode 100644
index b44e1f74d7..0000000000
--- a/docs/source/api/openfl_utilities.rst
+++ /dev/null
@@ -1,16 +0,0 @@
-.. # Copyright (C) 2020-2024 Intel Corporation
-.. # SPDX-License-Identifier: Apache-2.0
-
-*************************************************
-Utilities Module
-*************************************************
-
-Utilities modules reference:
-
-.. autosummary::
-   :toctree: _autosummary
-   :template: custom-module-template.rst
-   :recursive:
-
-   openfl.utilities
-   
\ No newline at end of file
diff --git a/docs/source/openfl/director_workflow.svg b/docs/source/openfl/director_workflow.svg
deleted file mode 100644
index bdc57e11d0..0000000000
--- a/docs/source/openfl/director_workflow.svg
+++ /dev/null
@@ -1 +0,0 @@
-<svg width="1280" height="720" xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" overflow="hidden"><defs><clipPath id="clip0"><rect x="0" y="0" width="1280" height="720"/></clipPath></defs><g clip-path="url(#clip0)"><rect x="0" y="0" width="1280" height="720" fill="#FFFFFF"/><rect x="363" y="226" width="553" height="469" fill="#5B9BD5" fill-opacity="0.2"/><text fill="#7F7F7F" font-family="Calibri,Calibri_MSFontService,sans-serif" font-weight="400" font-size="13" transform="translate(372.832 686)">OpenFL</text><path d="M402.5 262.5C402.5 255.873 407.873 250.5 414.5 250.5L532.5 250.5C539.127 250.5 544.5 255.873 544.5 262.5L544.5 310.5C544.5 317.127 539.127 322.5 532.5 322.5L414.5 322.5C407.873 322.5 402.5 317.127 402.5 310.5Z" stroke="#41719C" stroke-width="1.33333" stroke-miterlimit="8" fill="#5B9BD5" fill-rule="evenodd"/><text fill="#FFFFFF" font-family="Calibri,Calibri_MSFontService,sans-serif" font-weight="400" font-size="21" transform="translate(425.454 281)">Python API <tspan font-size="21" x="-2.19333" y="26">component</tspan></text><path d="M402.5 108.308C402.5 90.1889 417.189 75.5001 435.308 75.5001L511.692 75.5001C529.811 75.5001 544.5 90.1889 544.5 108.308L544.5 115.692C544.5 133.811 529.811 148.5 511.692 148.5L435.308 148.5C417.189 148.5 402.5 133.811 402.5 115.692Z" stroke="#507E32" stroke-width="1.33333" stroke-miterlimit="8" fill="#70AD47" fill-rule="evenodd"/><text fill="#FFFFFF" font-family="Calibri,Calibri_MSFontService,sans-serif" font-weight="400" font-size="19" transform="translate(429.324 106)">Experiment <tspan font-size="19" x="9.57333" y="23">Manager</tspan></text><path d="M442.5 57.0001C442.5 39.6031 456.379 25.5001 473.5 25.5001 490.621 25.5001 504.5 39.6031 504.5 57.0001 504.5 74.397 490.621 88.5001 473.5 88.5001 456.379 88.5001 442.5 74.397 442.5 57.0001Z" stroke="#507E32" stroke-width="1.33333" stroke-miterlimit="8" fill="#70AD47" fill-rule="evenodd"/><path d="M77.5001 459.859C77.5001 441.988 91.9876 427.5 109.859 427.5L186.141 427.5C204.012 427.5 218.5 441.988 218.5 459.859L218.5 467.141C218.5 485.012 204.012 499.5 186.141 499.5L109.859 499.5C91.9876 499.5 77.5001 485.012 77.5001 467.141Z" stroke="#507E32" stroke-width="1.33333" stroke-miterlimit="8" fill="#70AD47" fill-rule="evenodd"/><text fill="#FFFFFF" font-family="Calibri,Calibri_MSFontService,sans-serif" font-weight="400" font-size="19" transform="translate(116.438 458)">Director <tspan font-size="19" x="-3.30667" y="23">Manager</tspan></text><path d="M116.5 409C116.5 391.603 130.603 377.5 148 377.5 165.397 377.5 179.5 391.603 179.5 409 179.5 426.397 165.397 440.5 148 440.5 130.603 440.5 116.5 426.397 116.5 409Z" stroke="#507E32" stroke-width="1.33333" stroke-miterlimit="8" fill="#70AD47" fill-rule="evenodd"/><path d="M1062.5 461.309C1062.5 443.189 1077.19 428.5 1095.31 428.5L1170.69 428.5C1188.81 428.5 1203.5 443.189 1203.5 461.309L1203.5 468.692C1203.5 486.811 1188.81 501.5 1170.69 501.5L1095.31 501.5C1077.19 501.5 1062.5 486.811 1062.5 468.692Z" stroke="#507E32" stroke-width="1.33333" stroke-miterlimit="8" fill="#70AD47" fill-rule="evenodd"/><text fill="#FFFFFF" font-family="Calibri,Calibri_MSFontService,sans-serif" font-weight="400" font-size="19" transform="translate(1085.4 459)">Collaborator <tspan font-size="19" x="13.1267" y="23">Manager</tspan></text><path d="M1101.5 410.5C1101.5 393.379 1115.6 379.5 1133 379.5 1150.4 379.5 1164.5 393.379 1164.5 410.5 1164.5 427.621 1150.4 441.5 1133 441.5 1115.6 441.5 1101.5 427.621 1101.5 410.5Z" stroke="#507E32" stroke-width="1.33333" stroke-miterlimit="8" fill="#70AD47" fill-rule="evenodd"/><rect x="388" y="403" width="167" height="254" fill="#70AD47" fill-opacity="0.501961"/><text fill="#7F7F7F" font-family="Calibri,Calibri_MSFontService,sans-serif" font-weight="400" font-size="12" transform="translate(397.693 649)">Central node</text><path d="M400.5 440.5C400.5 433.873 405.873 428.5 412.5 428.5L529.5 428.5C536.127 428.5 541.5 433.873 541.5 440.5L541.5 488.5C541.5 495.127 536.127 500.5 529.5 500.5L412.5 500.5C405.873 500.5 400.5 495.127 400.5 488.5Z" stroke="#41719C" stroke-width="1.33333" stroke-miterlimit="8" fill="#5B9BD5" fill-rule="evenodd"/><text fill="#FFFFFF" font-family="Calibri,Calibri_MSFontService,sans-serif" font-weight="400" font-size="21" transform="translate(435.081 471)">Director</text><rect x="723" y="407" width="167" height="253" fill="#70AD47" fill-opacity="0.501961"/><text fill="#7F7F7F" font-family="Calibri,Calibri_MSFontService,sans-serif" font-weight="400" font-size="12" transform="translate(732.783 652)">Collaborator node</text><path d="M736.5 439.667C736.5 432.947 741.947 427.5 748.667 427.5L866.333 427.5C873.053 427.5 878.5 432.947 878.5 439.667L878.5 488.333C878.5 495.053 873.053 500.5 866.333 500.5L748.667 500.5C741.947 500.5 736.5 495.053 736.5 488.333Z" stroke="#41719C" stroke-width="1.33333" stroke-miterlimit="8" fill="#5B9BD5" fill-rule="evenodd"/><text fill="#FFFFFF" font-family="Calibri,Calibri_MSFontService,sans-serif" font-weight="400" font-size="21" transform="translate(781.363 470)">Envoy</text><path d="M400.5 576.5C400.5 569.873 405.873 564.5 412.5 564.5L529.5 564.5C536.127 564.5 541.5 569.873 541.5 576.5L541.5 624.5C541.5 631.127 536.127 636.5 529.5 636.5L412.5 636.5C405.873 636.5 400.5 631.127 400.5 624.5Z" stroke="#41719C" stroke-width="1.33333" stroke-miterlimit="8" fill="#5B9BD5" fill-rule="evenodd"/><text fill="#FFFFFF" font-family="Calibri,Calibri_MSFontService,sans-serif" font-weight="400" font-size="21" transform="translate(422.856 607)">Aggregator</text><path d="M736.5 576.5C736.5 569.873 741.873 564.5 748.5 564.5L866.5 564.5C873.127 564.5 878.5 569.873 878.5 576.5L878.5 624.5C878.5 631.127 873.127 636.5 866.5 636.5L748.5 636.5C741.873 636.5 736.5 631.127 736.5 624.5Z" stroke="#41719C" stroke-width="1.33333" stroke-miterlimit="8" fill="#5B9BD5" fill-rule="evenodd"/><text fill="#FFFFFF" font-family="Calibri,Calibri_MSFontService,sans-serif" font-weight="400" font-size="21" transform="translate(753.286 607)">Collaborator</text><path d="M0.333333-1.02799e-06 0.333628 95.462-0.333039 95.462-0.333333 1.02799e-06ZM4.00029 94.1286 0.000314961 102.129-3.99971 94.1286Z" fill="#70AD47" transform="matrix(-1 0 0 1 473.5 148.5)"/><rect x="385" y="163" width="183" height="48" fill="#FFFFFF"/><text fill="#7F7F7F" font-family="Calibri,Calibri_MSFontService,sans-serif" font-weight="400" font-size="16" transform="translate(398.235 183)">Provides FL plans, Tasks, <tspan font-size="16" x="8.10666" y="19">Models, Data Loaders</tspan></text><path d="M218.501 463.167 393.452 463.846 393.449 464.513 218.499 463.833ZM392.132 460.175 400.117 464.206 392.101 468.174Z" fill="#70AD47"/><rect x="240" y="419" width="116" height="88" fill="#FFFFFF"/><text fill="#7F7F7F" font-family="Calibri,Calibri_MSFontService,sans-serif" font-weight="400" font-size="16" transform="translate(265.803 439)">Launches.<tspan font-size="16" x="-12.8733" y="19">Sets up global </tspan><tspan font-size="16" x="-2.72668" y="38">Federation </tspan><tspan font-size="16" x="6.99332" y="58">settings</tspan></text><path d="M1062.5 465.173 885.164 463.882 885.169 463.215 1062.5 464.506ZM886.471 467.558 878.5 463.5 886.529 459.558Z" fill="#70AD47"/><rect x="921" y="419" width="131" height="88" fill="#FFFFFF"/><text fill="#7F7F7F" font-family="Calibri,Calibri_MSFontService,sans-serif" font-weight="400" font-size="16" transform="translate(954.221 439)">Launches.<tspan font-size="16" x="-12.4333" y="19">Provides local </tspan><tspan font-size="16" x="-12.18" y="38">dataset Shard </tspan><tspan font-size="16" x="-4.58667" y="58">Descriptors</tspan></text><path d="M471.833 549.5 471.833 557.654 471.167 557.654 471.167 549.5ZM475.5 556.321 471.5 564.321 467.5 556.321Z" fill="#5B9BD5"/><text fill="#7F7F7F" font-family="Calibri,Calibri_MSFontService,sans-serif" font-weight="400" font-size="13" transform="translate(410.583 524)">Creates an instance to <tspan font-size="13" x="-12.1667" y="16">maintain an FL experiment</tspan></text><path d="M471.5 499.5 471.5 512.274" stroke="#5B9BD5" stroke-width="0.666667" stroke-miterlimit="8" fill="none" fill-rule="evenodd"/><text fill="#7F7F7F" font-family="Calibri,Calibri_MSFontService,sans-serif" font-weight="400" font-size="13" transform="translate(747.029 527)">Creates an instance to <tspan font-size="13" x="-12.1667" y="16">maintain an FL experiment</tspan></text><path d="M0 0 0.000209974 11.1366" stroke="#5B9BD5" stroke-width="0.666667" stroke-miterlimit="8" fill="none" fill-rule="evenodd" transform="matrix(-1 0 0 1 807.5 500.5)"/><path d="M807.833 553.5 807.833 557.97 807.167 557.97 807.167 553.5ZM811.5 556.637 807.5 564.637 803.5 556.637Z" fill="#5B9BD5"/><text fill="#7F7F7F" font-family="Calibri,Calibri_MSFontService,sans-serif" font-weight="400" font-size="13" transform="translate(561.516 436)">Communicates dataset info, <tspan font-size="13" x="18.6666" y="16">Sends status updates</tspan><tspan font-size="13" x="25.5564" y="49">Approves, Sends FL </tspan><tspan font-size="13" x="43.8031" y="65">experiments</tspan><tspan font-size="13" x="2.12366" y="139">Sends locally tuned tensors </tspan><tspan font-size="13" x="21.5436" y="155">and training metrics</tspan><tspan font-size="13" x="18.4586" y="182">Sends tasks and initial </tspan><tspan font-size="13" x="57.5386" y="198">tensors</tspan></text><path d="M548.167 440.167 553.5 440.167 553.5 440.833 548.167 440.833ZM555.5 440.167 560.833 440.167 560.833 440.833 555.5 440.833ZM562.833 440.167 568.167 440.167 568.167 440.833 562.833 440.833ZM570.167 440.167 575.5 440.167 575.5 440.833 570.167 440.833ZM577.5 440.167 582.833 440.167 582.833 440.833 577.5 440.833ZM584.833 440.167 590.167 440.167 590.167 440.833 584.833 440.833ZM592.167 440.167 597.5 440.167 597.5 440.833 592.167 440.833ZM599.5 440.167 604.833 440.167 604.833 440.833 599.5 440.833ZM606.833 440.167 612.167 440.167 612.167 440.833 606.833 440.833ZM614.167 440.167 619.5 440.167 619.5 440.833 614.167 440.833ZM621.5 440.167 626.833 440.167 626.833 440.833 621.5 440.833ZM628.833 440.167 634.167 440.167 634.167 440.833 628.833 440.833ZM636.167 440.167 641.5 440.167 641.5 440.833 636.167 440.833ZM643.5 440.167 648.833 440.167 648.833 440.833 643.5 440.833ZM650.833 440.167 656.167 440.167 656.167 440.833 650.833 440.833ZM658.167 440.167 663.5 440.167 663.5 440.833 658.167 440.833ZM665.5 440.167 670.833 440.167 670.833 440.833 665.5 440.833ZM672.833 440.167 678.167 440.167 678.167 440.833 672.833 440.833ZM680.167 440.167 685.5 440.167 685.5 440.833 680.167 440.833ZM687.5 440.167 692.833 440.167 692.833 440.833 687.5 440.833ZM694.833 440.167 700.167 440.167 700.167 440.833 694.833 440.833ZM702.167 440.167 707.5 440.167 707.5 440.833 702.167 440.833ZM709.5 440.167 714.833 440.167 714.833 440.833 709.5 440.833ZM716.833 440.167 722.167 440.167 722.167 440.833 716.833 440.833ZM724.167 440.167 729.5 440.167 729.5 440.833 724.167 440.833ZM731.5 440.167 736.118 440.167 736.118 440.833 731.5 440.833ZM549.5 444.5 541.5 440.5 549.5 436.5Z" fill="#5B9BD5"/><path d="M6.6667-0.33333 12-0.333327 12 0.33334 6.6667 0.333337ZM14-0.333326 19.3334-0.333323 19.3334 0.333344 14 0.333341ZM21.3334-0.333322 26.6667-0.333319 26.6667 0.333348 21.3334 0.333345ZM28.6667-0.333318 34-0.333315 34 0.333352 28.6667 0.333349ZM36-0.333314 41.3334-0.333311 41.3334 0.333356 36 0.333353ZM43.3334-0.33331 48.6667-0.333307 48.6667 0.33336 43.3334 0.333357ZM50.6667-0.333306 56-0.333303 56 0.333364 50.6667 0.333361ZM58-0.333302 63.3334-0.333299 63.3334 0.333367 58 0.333365ZM65.3334-0.333298 70.6667-0.333295 70.6667 0.333371 65.3334 0.333369ZM72.6667-0.333294 78-0.333291 78 0.333375 72.6667 0.333373ZM80-0.33329 85.3334-0.333287 85.3334 0.333379 80 0.333376ZM87.3334-0.333286 92.6667-0.333283 92.6667 0.333383 87.3334 0.33338ZM94.6667-0.333282 100-0.333279 100 0.333387 94.6667 0.333384ZM102-0.333278 107.333-0.333275 107.333 0.333391 102 0.333388ZM109.333-0.333274 114.667-0.333271 114.667 0.333395 109.333 0.333392ZM116.667-0.33327 122-0.333268 122 0.333399 116.667 0.333396ZM124-0.333266 129.333-0.333264 129.333 0.333403 124 0.3334ZM131.333-0.333262 136.667-0.33326 136.667 0.333407 131.333 0.333404ZM138.667-0.333259 144-0.333256 144 0.333411 138.667 0.333408ZM146-0.333255 151.333-0.333252 151.333 0.333415 146 0.333412ZM153.333-0.333251 158.667-0.333248 158.667 0.333419 153.333 0.333416ZM160.667-0.333247 166-0.333244 166 0.333423 160.667 0.33342ZM168-0.333243 173.333-0.33324 173.333 0.333427 168 0.333424ZM175.333-0.333239 180.667-0.333236 180.667 0.333431 175.333 0.333428ZM182.667-0.333235 188-0.333232 188 0.333435 182.667 0.333432ZM190-0.333231 194.618-0.333228 194.618 0.333438 190 0.333436ZM8 4 0 0 8-4Z" fill="#5B9BD5" transform="matrix(-1 0 0 1 736.118 491.5)"/><path d="M548.167 581.167 553.5 581.167 553.5 581.833 548.167 581.833ZM555.5 581.167 560.833 581.167 560.833 581.833 555.5 581.833ZM562.833 581.167 568.167 581.167 568.167 581.833 562.833 581.833ZM570.167 581.167 575.5 581.167 575.5 581.833 570.167 581.833ZM577.5 581.167 582.833 581.167 582.833 581.833 577.5 581.833ZM584.833 581.167 590.167 581.167 590.167 581.833 584.833 581.833ZM592.167 581.167 597.5 581.167 597.5 581.833 592.167 581.833ZM599.5 581.167 604.833 581.167 604.833 581.833 599.5 581.833ZM606.833 581.167 612.167 581.167 612.167 581.833 606.833 581.833ZM614.167 581.167 619.5 581.167 619.5 581.833 614.167 581.833ZM621.5 581.167 626.833 581.167 626.833 581.833 621.5 581.833ZM628.833 581.167 634.167 581.167 634.167 581.833 628.833 581.833ZM636.167 581.167 641.5 581.167 641.5 581.833 636.167 581.833ZM643.5 581.167 648.833 581.167 648.833 581.833 643.5 581.833ZM650.833 581.167 656.167 581.167 656.167 581.833 650.833 581.833ZM658.167 581.167 663.5 581.167 663.5 581.833 658.167 581.833ZM665.5 581.167 670.833 581.167 670.833 581.833 665.5 581.833ZM672.833 581.167 678.167 581.167 678.167 581.833 672.833 581.833ZM680.167 581.167 685.5 581.167 685.5 581.833 680.167 581.833ZM687.5 581.167 692.833 581.167 692.833 581.833 687.5 581.833ZM694.833 581.167 700.167 581.167 700.167 581.833 694.833 581.833ZM702.167 581.167 707.5 581.167 707.5 581.833 702.167 581.833ZM709.5 581.167 714.833 581.167 714.833 581.833 709.5 581.833ZM716.833 581.167 722.167 581.167 722.167 581.833 716.833 581.833ZM724.167 581.167 729.5 581.167 729.5 581.833 724.167 581.833ZM731.5 581.167 736.118 581.167 736.118 581.833 731.5 581.833ZM549.5 585.5 541.5 581.5 549.5 577.5Z" fill="#5B9BD5"/><path d="M6.6667-0.33333 12-0.333327 12 0.33334 6.6667 0.333337ZM14-0.333326 19.3334-0.333323 19.3334 0.333344 14 0.333341ZM21.3334-0.333322 26.6667-0.333319 26.6667 0.333348 21.3334 0.333345ZM28.6667-0.333318 34-0.333315 34 0.333352 28.6667 0.333349ZM36-0.333314 41.3334-0.333311 41.3334 0.333356 36 0.333353ZM43.3334-0.33331 48.6667-0.333307 48.6667 0.33336 43.3334 0.333357ZM50.6667-0.333306 56-0.333303 56 0.333364 50.6667 0.333361ZM58-0.333302 63.3334-0.333299 63.3334 0.333367 58 0.333365ZM65.3334-0.333298 70.6667-0.333295 70.6667 0.333371 65.3334 0.333369ZM72.6667-0.333294 78-0.333291 78 0.333375 72.6667 0.333373ZM80-0.33329 85.3334-0.333287 85.3334 0.333379 80 0.333376ZM87.3334-0.333286 92.6667-0.333283 92.6667 0.333383 87.3334 0.33338ZM94.6667-0.333282 100-0.333279 100 0.333387 94.6667 0.333384ZM102-0.333278 107.333-0.333275 107.333 0.333391 102 0.333388ZM109.333-0.333274 114.667-0.333271 114.667 0.333395 109.333 0.333392ZM116.667-0.33327 122-0.333268 122 0.333399 116.667 0.333396ZM124-0.333266 129.333-0.333264 129.333 0.333403 124 0.3334ZM131.333-0.333262 136.667-0.33326 136.667 0.333407 131.333 0.333404ZM138.667-0.333259 144-0.333256 144 0.333411 138.667 0.333408ZM146-0.333255 151.333-0.333252 151.333 0.333415 146 0.333412ZM153.333-0.333251 158.667-0.333248 158.667 0.333419 153.333 0.333416ZM160.667-0.333247 166-0.333244 166 0.333423 160.667 0.33342ZM168-0.333243 173.333-0.33324 173.333 0.333427 168 0.333424ZM175.333-0.333239 180.667-0.333236 180.667 0.333431 175.333 0.333428ZM182.667-0.333235 188-0.333232 188 0.333435 182.667 0.333432ZM190-0.333231 194.618-0.333228 194.618 0.333438 190 0.333436ZM8 4 0 0 8-4Z" fill="#5B9BD5" transform="matrix(-1 0 0 1 736.118 624.5)"/><path d="M429.833 390.5 429.833 395.833 429.167 395.833 429.167 390.5ZM429.833 397.833 429.833 403.167 429.167 403.167 429.167 397.833ZM429.833 405.167 429.833 410.5 429.167 410.5 429.167 405.167ZM429.833 412.5 429.833 417.833 429.167 417.833 429.167 412.5ZM429.833 419.833 429.833 422.069 429.167 422.069 429.167 419.833ZM433.5 420.736 429.5 428.736 425.5 420.736Z" fill="#5B9BD5"/><path d="M0.333333-2.36129e-06 0.333371 5.33333-0.333296 5.33334-0.333333 2.36129e-06ZM0.333385 7.33333 0.333391 8.15391-0.333276 8.15391-0.333281 7.33334ZM4.00005 6.82055 0.000104987 14.8206-3.99995 6.82061Z" fill="#5B9BD5" transform="matrix(1 0 0 -1 516.5 337.321)"/><text fill="#7F7F7F" font-family="Calibri,Calibri_MSFontService,sans-serif" font-weight="400" font-size="13" transform="translate(396.932 365)">Registers FL <tspan font-size="13" x="-1.83331" y="16">experiments</tspan><tspan font-size="13" x="73.5058" y="-16">Sends info about </tspan><tspan font-size="13" x="77.1725" y="0">the Federation. </tspan><tspan font-size="13" x="75.4192" y="16">Returns training </tspan><tspan font-size="13" x="96.8392" y="32">artifacts</tspan></text><path d="M429.5 322.5 429.5 345.939" stroke="#5B9BD5" stroke-width="0.666667" stroke-miterlimit="8" stroke-dasharray="5.33333 2" fill="none" fill-rule="evenodd"/><path d="M516.5 407.5 516.5 429.298" stroke="#5B9BD5" stroke-width="0.666667" stroke-miterlimit="8" stroke-dasharray="5.33333 2" fill="none" fill-rule="evenodd"/></g></svg>
\ No newline at end of file
diff --git a/docs/source/openfl/static_diagram.svg b/docs/source/openfl/static_diagram.svg
deleted file mode 100644
index 8f816d4e88..0000000000
--- a/docs/source/openfl/static_diagram.svg
+++ /dev/null
@@ -1 +0,0 @@
-<svg xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" version="1.1" viewBox="0 0 3104 2546" style="background: #ffffff"><g id="v-3"  transform="matrix(1,0,0,1,0,0)"><g id="j_36"    transform="translate(20,2501)"><g id="v-667"><text  id="v-668" font-size="22px" y="0.8em" xml:space="preserve" font-weight="normal" text-anchor="start" fill="#aaaaaa" pointer-events="none" font-family="Open Sans" transform="matrix(1,0,0,1,0,0)"><tspan id="v-669"  dy="0em" x="0">Friday, 27 August 2021, 16:25 Moscow Standard Time</tspan></text></g></g><g id="j_35"    transform="translate(20,2455)"><g id="v-663"><text  id="v-664" font-size="36px" y="0.8em" xml:space="preserve" font-weight="normal" text-anchor="start" fill="#000000" pointer-events="none" font-family="Open Sans" transform="matrix(1,0,0,1,0,0)"><tspan id="v-665"  dy="0em" x="0">Container diagram for OpenFL</tspan></text></g></g><g id="j_13"    transform="translate(840,740)" style=""><g id="v-214"><rect  id="v-215" width="1390" height="1582" rx="0" ry="0" fill="#ffffff" stroke="#444444" stroke-width="2" stroke-dasharray="20,20" pointer-events="none"></rect><text  id="v-216" font-size="24px" y="1543" xml:space="preserve" font-weight="normal" fill="#444444" x="10" text-anchor="start" pointer-events="visible" font-family="Open Sans" transform="matrix(1,0,0,1,15,0)"><tspan id="v-218"  dy="0em" x="0">OpenFL</tspan></text><text  id="v-217" font-size="19px" y="1567" xml:space="preserve" font-weight="normal" fill="#444444" x="10" text-anchor="start" pointer-events="visible" font-family="Open Sans" transform="matrix(1,0,0,1,15,0)"><tspan id="v-219"  dy="0em" x="0">[Software System]</tspan></text></g></g><g id="j_22"    transform="translate(860,1300)" style=""><g id="v-342"><rect  id="v-343" width="490" height="929" rx="0" ry="0" fill="#ffffff" stroke="#cccccc" stroke-width="2" stroke-dasharray="5,5" pointer-events="none"></rect><text  id="v-344" font-size="24px" y="914" xml:space="preserve" font-weight="normal" fill="#cccccc" x="10" text-anchor="start" pointer-events="visible" font-family="Open Sans" transform="matrix(1,0,0,1,15,0)"><tspan id="v-346"  dy="0em" x="0">Central node</tspan></text><text  id="v-345" font-size="19px" y="914" display="none" xml:space="preserve" font-weight="normal" fill="#cccccc" x="10" text-anchor="start" pointer-events="visible" font-family="Open Sans" transform="matrix(1,0,0,1,15,0)"><tspan id="v-347"  dy="0em" x="0" style="fill-opacity: 0; stroke-opacity: 0;">-</tspan></text></g></g><g id="j_16"    transform="translate(1720,1300)" style=""><g id="v-253"><rect  id="v-254" width="490" height="929" rx="0" ry="0" fill="#ffffff" stroke="#cccccc" stroke-width="2" stroke-dasharray="5,5" pointer-events="none"></rect><text  id="v-255" font-size="24px" y="914" xml:space="preserve" font-weight="normal" fill="#cccccc" x="10" text-anchor="start" pointer-events="visible" font-family="Open Sans" transform="matrix(1,0,0,1,15,0)"><tspan id="v-257"  dy="0em" x="0">Collaborator node</tspan></text><text  id="v-256" font-size="19px" y="914" display="none" xml:space="preserve" font-weight="normal" fill="#cccccc" x="10" text-anchor="start" pointer-events="visible" font-family="Open Sans" transform="matrix(1,0,0,1,15,0)"><tspan id="v-258"  dy="0em" x="0" style="fill-opacity: 0; stroke-opacity: 0;">-</tspan></text></g></g><g id="j_14"    transform="translate(890,200)" style=""><g  id="v-223" style="opacity: 1;"><rect  x="0" y="160" width="400" height="240" rx="70" id="v-225" stroke="#073b6f" stroke-width="2" pointer-events="visiblePainted" fill="#08427b"></rect><circle  cx="200" cy="88.88888888888889" r="88.88888888888889" id="v-224" stroke="#073b6f" stroke-width="2" pointer-events="visiblePainted" fill="#08427b"></circle><line  x1="80" y1="266.6666666666667" x2="80" y2="400" style="stroke-width:2px" id="v-231" stroke="#073b6f"></line><line  x1="320" y1="266.6666666666667" x2="320" y2="400" style="stroke-width:2px" id="v-232" stroke="#073b6f"></line><text  id="v-226" font-size="34px" y="0.8em" xml:space="preserve" font-weight="bold" text-anchor="middle" pointer-events="visible" font-family="Open Sans" fill="#ffffff" transform="matrix(1,0,0,1,200,205)"><tspan id="v-233"  dy="0em" x="0">Data scientist</tspan></text><text  id="v-227" font-size="19px" y="0.8em" xml:space="preserve" text-anchor="middle" font-family="Open Sans" fill="#ffffff" transform="matrix(1,0,0,1,200,247.5)"><tspan id="v-234"  dy="0em" x="0">[Person]</tspan></text><text  id="v-228" font-size="24px" y="0.8em" xml:space="preserve" text-anchor="middle" font-family="Open Sans" fill="#ffffff" transform="matrix(1,0,0,1,200,295)"><tspan id="v-235"  dy="0em" x="0">A person or group of people</tspan><tspan id="v-236"  dy="1.2em" x="0">using OpenFL</tspan></text><text  id="v-229" font-weight="bold" text-anchor="middle" font-family="Open Sans" fill="#ffffff" font-size="24" transform="matrix(1,0,0,1,200,371.2)" display="none"></text><image  id="v-230"></image></g></g><g id="j_15"    transform="translate(1740,1320)" style="" ><g  id="v-240" style="opacity: 1;"><rect  id="v-241" rx="20" ry="20" stroke="#3c7fc0" stroke-width="2" pointer-events="visiblePainted" fill="#438dd5" width="450" height="300"></rect><text  id="v-242" font-size="34" y="0.8em" xml:space="preserve" font-weight="bold" text-anchor="middle" pointer-events="visible" font-family="Open Sans" fill="#ffffff" transform="matrix(1,0,0,1,225,60)"><tspan id="v-247"  dy="0em" x="0">Envoy</tspan></text><text  id="v-243" font-size="19" y="0.8em" xml:space="preserve" text-anchor="middle" font-family="Open Sans" fill="#ffffff" transform="matrix(1,0,0,1,225,102.5)"><tspan id="v-248"  dy="0em" x="0">[Container]</tspan></text><text  id="v-244" font-size="24" y="0.8em" xml:space="preserve" text-anchor="middle" font-family="Open Sans" fill="#ffffff" transform="matrix(1,0,0,1,225,150)"><tspan id="v-249"  dy="0em" x="0">A long-living entity that can adapt a</tspan><tspan id="v-250"  dy="1.2em" x="0">local data set and spawn</tspan><tspan id="v-251"  dy="1.2em" x="0">collaborators</tspan></text><text  id="v-245" font-weight="bold" text-anchor="middle" font-family="Open Sans" fill="#ffffff" font-size="24" transform="matrix(1,0,0,1,225,270)" y="0.8em" xml:space="preserve" display="none"><tspan id="v-259"  dy="0em" x="0">+</tspan></text><image  id="v-246"></image></g></g><g id="j_17"    transform="translate(2470,1265)" style=""><g  id="v-265" style="opacity: 1;"><rect  x="0" y="160" width="400" height="240" rx="70" id="v-267" stroke="#073b6f" stroke-width="2" pointer-events="visiblePainted" fill="#08427b"></rect><circle  cx="200" cy="88.88888888888889" r="88.88888888888889" id="v-266" stroke="#073b6f" stroke-width="2" pointer-events="visiblePainted" fill="#08427b"></circle><line  x1="80" y1="266.6666666666667" x2="80" y2="400" style="stroke-width:2px" id="v-273" stroke="#073b6f"></line><line  x1="320" y1="266.6666666666667" x2="320" y2="400" style="stroke-width:2px" id="v-274" stroke="#073b6f"></line><text  id="v-268" font-size="34px" y="0.8em" xml:space="preserve" font-weight="bold" text-anchor="middle" pointer-events="visible" font-family="Open Sans" fill="#ffffff" transform="matrix(1,0,0,1,200,205)"><tspan id="v-275"  dy="0em" x="0">Collaborator manager</tspan></text><text  id="v-269" font-size="19px" y="0.8em" xml:space="preserve" text-anchor="middle" font-family="Open Sans" fill="#ffffff" transform="matrix(1,0,0,1,200,247.5)"><tspan id="v-276"  dy="0em" x="0">[Person]</tspan></text><text  id="v-270" font-size="24px" y="0.8em" xml:space="preserve" text-anchor="middle" font-family="Open Sans" fill="#ffffff" transform="matrix(1,0,0,1,200,295)"><tspan id="v-277"  dy="0em" x="0">Data owner's representative</tspan><tspan id="v-278"  dy="1.2em" x="0">controlling Envoy</tspan></text><text  id="v-271" font-weight="bold" text-anchor="middle" font-family="Open Sans" fill="#ffffff" font-size="24" transform="matrix(1,0,0,1,200,371.2)" display="none"></text><image  id="v-272"></image></g></g><g id="j_18"    transform="translate(230,1270)" style=""><g  id="v-282" style="opacity: 1;"><rect  x="0" y="160" width="400" height="240" rx="70" id="v-284" stroke="#073b6f" stroke-width="2" pointer-events="visiblePainted" fill="#08427b"></rect><circle  cx="200" cy="88.88888888888889" r="88.88888888888889" id="v-283" stroke="#073b6f" stroke-width="2" pointer-events="visiblePainted" fill="#08427b"></circle><line  x1="80" y1="266.6666666666667" x2="80" y2="400" style="stroke-width:2px" id="v-290" stroke="#073b6f"></line><line  x1="320" y1="266.6666666666667" x2="320" y2="400" style="stroke-width:2px" id="v-291" stroke="#073b6f"></line><text  id="v-285" font-size="34px" y="0.8em" xml:space="preserve" font-weight="bold" text-anchor="middle" pointer-events="visible" font-family="Open Sans" fill="#ffffff" transform="matrix(1,0,0,1,200,246.9)"><tspan id="v-292"  dy="0em" x="0">Director manager</tspan></text><text  id="v-286" font-size="19px" y="0.8em" xml:space="preserve" text-anchor="middle" font-family="Open Sans" fill="#ffffff" transform="matrix(1,0,0,1,200,289.4)"><tspan id="v-293"  dy="0em" x="0">[Person]</tspan></text><text  id="v-287" font-size="24px" y="0.8em" display="none" xml:space="preserve" text-anchor="middle" font-family="Open Sans" fill="#ffffff" transform="matrix(1,0,0,1,200,313.1)"><tspan id="v-294"  dy="0em" x="0" style="fill-opacity: 0; stroke-opacity: 0;">-</tspan></text><text  id="v-288" font-weight="bold" text-anchor="middle" font-family="Open Sans" fill="#ffffff" font-size="24" transform="matrix(1,0,0,1,200,371.2)" display="none"></text><image  id="v-289"></image></g></g><g id="j_19"    transform="translate(1740,1855)" style="" ><g  id="v-298" style="opacity: 1;"><rect  id="v-299" rx="20" ry="20" stroke="#3c7fc0" stroke-width="2" pointer-events="visiblePainted" fill="#438dd5" width="450" height="300"></rect><text  id="v-300" font-size="34" y="0.8em" xml:space="preserve" font-weight="bold" text-anchor="middle" pointer-events="visible" font-family="Open Sans" fill="#ffffff" transform="matrix(1,0,0,1,225,75)"><tspan id="v-305"  dy="0em" x="0">Collaborator</tspan></text><text  id="v-301" font-size="19" y="0.8em" xml:space="preserve" text-anchor="middle" font-family="Open Sans" fill="#ffffff" transform="matrix(1,0,0,1,225,117.5)"><tspan id="v-306"  dy="0em" x="0">[Container]</tspan></text><text  id="v-302" font-size="24" y="0.8em" xml:space="preserve" text-anchor="middle" font-family="Open Sans" fill="#ffffff" transform="matrix(1,0,0,1,225,165)"><tspan id="v-307"  dy="0em" x="0">Actor executing tasks on local data</tspan><tspan id="v-308"  dy="1.2em" x="0">inside one experiment</tspan></text><text  id="v-303" font-weight="bold" text-anchor="middle" font-family="Open Sans" fill="#ffffff" font-size="24" transform="matrix(1,0,0,1,225,270)" y="0.8em" xml:space="preserve" display="none"><tspan id="v-309"  dy="0em" x="0">+</tspan></text><image  id="v-304"></image></g></g><g id="j_20"    transform="translate(880,760)" style="" ><g  id="v-315" style="opacity: 1;"><rect  id="v-316" rx="20" ry="20" stroke="#3c7fc0" stroke-width="2" pointer-events="visiblePainted" fill="#438dd5" width="450" height="300"></rect><text  id="v-317" font-size="34" y="0.8em" xml:space="preserve" font-weight="bold" text-anchor="middle" pointer-events="visible" font-family="Open Sans" fill="#ffffff" transform="matrix(1,0,0,1,225,75)"><tspan id="v-322"  dy="0em" x="0">Python API component</tspan></text><text  id="v-318" font-size="19" y="0.8em" xml:space="preserve" text-anchor="middle" font-family="Open Sans" fill="#ffffff" transform="matrix(1,0,0,1,225,117.5)"><tspan id="v-323"  dy="0em" x="0">[Container]</tspan></text><text  id="v-319" font-size="24" y="0.8em" xml:space="preserve" text-anchor="middle" font-family="Open Sans" fill="#ffffff" transform="matrix(1,0,0,1,225,165)"><tspan id="v-324"  dy="0em" x="0">A set of tools to setup register FL</tspan><tspan id="v-325"  dy="1.2em" x="0">Experiments</tspan></text><text  id="v-320" font-weight="bold" text-anchor="middle" font-family="Open Sans" fill="#ffffff" font-size="24" transform="matrix(1,0,0,1,225,270)" y="0.8em" xml:space="preserve" display="none"><tspan id="v-326"  dy="0em" x="0">+</tspan></text><image  id="v-321"></image></g></g><g id="j_21"    transform="translate(880,1320)" style="" ><g  id="v-330" style="opacity: 1;"><rect  id="v-331" rx="20" ry="20" stroke="#3c7fc0" stroke-width="2" pointer-events="visiblePainted" fill="#438dd5" width="450" height="300"></rect><text  id="v-332" font-size="34" y="0.8em" xml:space="preserve" font-weight="bold" text-anchor="middle" pointer-events="visible" font-family="Open Sans" fill="#ffffff" transform="matrix(1,0,0,1,225,75)"><tspan id="v-337"  dy="0em" x="0">Director</tspan></text><text  id="v-333" font-size="19" y="0.8em" xml:space="preserve" text-anchor="middle" font-family="Open Sans" fill="#ffffff" transform="matrix(1,0,0,1,225,117.5)"><tspan id="v-338"  dy="0em" x="0">[Container]</tspan></text><text  id="v-334" font-size="24" y="0.8em" xml:space="preserve" text-anchor="middle" font-family="Open Sans" fill="#ffffff" transform="matrix(1,0,0,1,225,165)"><tspan id="v-339"  dy="0em" x="0">A long-living entity that can spawn</tspan><tspan id="v-340"  dy="1.2em" x="0">aggregators</tspan></text><text  id="v-335" font-weight="bold" text-anchor="middle" font-family="Open Sans" fill="#ffffff" font-size="24" transform="matrix(1,0,0,1,225,270)" y="0.8em" display="none" xml:space="preserve"><tspan id="v-348"  dy="0em" x="0" style="fill-opacity: 0; stroke-opacity: 0;">-</tspan></text><image  id="v-336"></image></g></g><g id="j_23"    transform="translate(880,1855)" style="" ><g  id="v-352" style="opacity: 1;"><rect  id="v-353" rx="20" ry="20" stroke="#3c7fc0" stroke-width="2" pointer-events="visiblePainted" fill="#438dd5" width="450" height="300"></rect><text  id="v-354" font-size="34" y="0.8em" xml:space="preserve" font-weight="bold" text-anchor="middle" pointer-events="visible" font-family="Open Sans" fill="#ffffff" transform="matrix(1,0,0,1,225,75)"><tspan id="v-359"  dy="0em" x="0">Aggregator</tspan></text><text  id="v-355" font-size="19" y="0.8em" xml:space="preserve" text-anchor="middle" font-family="Open Sans" fill="#ffffff" transform="matrix(1,0,0,1,225,117.5)"><tspan id="v-360"  dy="0em" x="0">[Container]</tspan></text><text  id="v-356" font-size="24" y="0.8em" xml:space="preserve" text-anchor="middle" font-family="Open Sans" fill="#ffffff" transform="matrix(1,0,0,1,225,165)"><tspan id="v-361"  dy="0em" x="0">Model server and collaborator</tspan><tspan id="v-362"  dy="1.2em" x="0">orchestrator</tspan></text><text  id="v-357" font-weight="bold" text-anchor="middle" font-family="Open Sans" fill="#ffffff" font-size="24" transform="matrix(1,0,0,1,225,270)" y="0.8em" display="none" xml:space="preserve"><tspan id="v-363"  dy="0em" x="0" style="fill-opacity: 0; stroke-opacity: 0;">-</tspan></text><image  id="v-358"></image></g></g><g id="j_24"   ><path  stroke="#707070" id="v-386" stroke-width="2" stroke-dasharray="30 30" fill="none" d="M 630 1470 860 1470" style="opacity: 1;"></path><path  fill="black" stroke="black" transform="translate(630,1470) scale(1,1) rotate(0)"></path><path  fill="#707070" stroke="#707070" id="v-388" d="M 20 0 L 0 10 L 20 20 z" transform="translate(880,1480) scale(1,1) rotate(-180)" style="opacity: 1;"></path><path  id="v-387" fill="none" d="M 630 1470 860 1470"></path><title ></title><g ><g  id="v-371" label-idx="0" cursor="move" transform="translate(745, 1470)" style="opacity: 1;"><rect id="v-373" fill="#ffffff" rx="3" ry="3" stroke="#ffffff" stroke-width="20px" pointer-events="none" width="165.5" height="119" transform="matrix(1,0,0,1,-82.7,-59.5)"></rect><text id="v-372" font-size="24px" y="0.8em" xml:space="preserve" text-anchor="middle" pointer-events="none" fill="#707070" font-family="Open Sans" font-weight="normal" transform="matrix(1,0,0,1,0,-52.2)"><tspan id="v-374"  dy="0em" x="0">Launches. Sets</tspan><tspan id="v-375"  dy="1.2em" x="0">up global</tspan><tspan id="v-376"  dy="1.2em" x="0">Federation</tspan><tspan id="v-377"  dy="1.2em" x="0">settings</tspan></text></g><g  id="v-378" label-idx="1" cursor="move" transform="translate(745, 1539.6)" style="opacity: 1;"><rect id="v-380" fill="#ffffff" rx="3" ry="3" pointer-events="none" width="0" height="0" transform="matrix(1,0,0,1,0,0)"></rect><text id="v-379" font-size="19px" y="0.8em" display="none" xml:space="preserve" text-anchor="middle" pointer-events="none" fill="#707070" font-family="Open Sans" font-weight="normal" transform="matrix(1,0,0,1,0,0)"><tspan id="v-381"  dy="0em" x="0" style="fill-opacity: 0; stroke-opacity: 0;">-</tspan></text></g><g  id="v-382" label-idx="2" cursor="move" transform="translate(745, 1563.6)" style="opacity: 1;"><rect id="v-384" fill="#ffffff" rx="3" ry="3" pointer-events="none"  display="none" width="0" height="0" transform="matrix(1,0,0,1,0,0)"></rect><text id="v-383" font-size="19px" y="0.8em" display="none" xml:space="preserve" text-anchor="middle" pointer-events="none" fill="#707070" font-family="Open Sans" font-weight="normal"  transform="matrix(1,0,0,1,0,0)"><tspan id="v-385"  dy="0em" x="0" style="fill-opacity: 0; stroke-opacity: 0;">-</tspan></text></g></g><g  display="none"></g><g  id="v-390" display="none"><g  id="v-368"><path  end="source" d="M 26 0 L 0 13 L 26 26 z"></path></g><g  id="v-369"><path  end="target" d="M 26 0 L 0 13 L 26 26 z"></path></g></g><g  id="v-389" display="none"><g  id="v-367" transform="translate(670, 1470) "><g  event="remove"><circle r="11"></circle><path transform="scale(.8) translate(-16, -16)" d="M24.778,21.419 19.276,15.917 24.777,10.415 21.949,7.585 16.447,13.087 10.945,7.585 8.117,10.415 13.618,15.917 8.116,21.419 10.946,24.248 16.447,18.746 21.948,24.248z"></path><title>Remove link.</title></g><g  event="link:options"><circle r="11" transform="translate(25)"></circle><path fill="white" transform="scale(.55) translate(29, -16)" d="M31.229,17.736c0.064-0.571,0.104-1.148,0.104-1.736s-0.04-1.166-0.104-1.737l-4.377-1.557c-0.218-0.716-0.504-1.401-0.851-2.05l1.993-4.192c-0.725-0.91-1.549-1.734-2.458-2.459l-4.193,1.994c-0.647-0.347-1.334-0.632-2.049-0.849l-1.558-4.378C17.165,0.708,16.588,0.667,16,0.667s-1.166,0.041-1.737,0.105L12.707,5.15c-0.716,0.217-1.401,0.502-2.05,0.849L6.464,4.005C5.554,4.73,4.73,5.554,4.005,6.464l1.994,4.192c-0.347,0.648-0.632,1.334-0.849,2.05l-4.378,1.557C0.708,14.834,0.667,15.412,0.667,16s0.041,1.165,0.105,1.736l4.378,1.558c0.217,0.715,0.502,1.401,0.849,2.049l-1.994,4.193c0.725,0.909,1.549,1.733,2.459,2.458l4.192-1.993c0.648,0.347,1.334,0.633,2.05,0.851l1.557,4.377c0.571,0.064,1.148,0.104,1.737,0.104c0.588,0,1.165-0.04,1.736-0.104l1.558-4.377c0.715-0.218,1.399-0.504,2.049-0.851l4.193,1.993c0.909-0.725,1.733-1.549,2.458-2.458l-1.993-4.193c0.347-0.647,0.633-1.334,0.851-2.049L31.229,17.736zM16,20.871c-2.69,0-4.872-2.182-4.872-4.871c0-2.69,2.182-4.872,4.872-4.872c2.689,0,4.871,2.182,4.871,4.872C20.871,18.689,18.689,20.871,16,20.871z"></path><title>Link options.</title></g></g></g></g><g id="j_25"   ><path  stroke="#707070" id="v-413" stroke-width="2" stroke-dasharray="30 30" fill="none" d="M 2470 1466 2210 1468" style="opacity: 1;"></path><path  fill="black" stroke="black" transform="translate(2470,1466) scale(1,1) rotate(-180.4092559814453)"></path><path  fill="#707070" stroke="#707070" id="v-415" d="M 20 0 L 0 10 L 20 20 z" transform="translate(2189.928584200193,1458.0002707109588) scale(1,1) rotate(-0.40924862027168274)" style="opacity: 1;"></path><path  id="v-414" fill="none" d="M 2470 1466 2210 1468"></path><title ></title><g ><g  id="v-398" label-idx="0" cursor="move" transform="translate(2340, 1467)" style="opacity: 1;"><rect id="v-400" fill="#ffffff" rx="3" ry="3" stroke="#ffffff" stroke-width="20px" pointer-events="none" width="194.34375" height="119" transform="matrix(1,0,0,1,-97.2,-59.5)"></rect><text id="v-399" font-size="24px" y="0.8em" xml:space="preserve" text-anchor="middle" pointer-events="none" fill="#707070" font-family="Open Sans" font-weight="normal" transform="matrix(1,0,0,1,0,-52.2)"><tspan id="v-401"  dy="0em" x="0">Launches.</tspan><tspan id="v-402"  dy="1.2em" x="0">Provides local</tspan><tspan id="v-403"  dy="1.2em" x="0">dataset</tspan><tspan id="v-404"  dy="1.2em" x="0">ShardDescriptors</tspan></text></g><g  id="v-405" label-idx="1" cursor="move" transform="translate(2340, 1536.6)" style="opacity: 1;"><rect id="v-407" fill="#ffffff" rx="3" ry="3" pointer-events="none" width="0" height="0" transform="matrix(1,0,0,1,0,0)"></rect><text id="v-406" font-size="19px" y="0.8em" display="none" xml:space="preserve" text-anchor="middle" pointer-events="none" fill="#707070" font-family="Open Sans" font-weight="normal" transform="matrix(1,0,0,1,0,0)"><tspan id="v-408"  dy="0em" x="0" style="fill-opacity: 0; stroke-opacity: 0;">-</tspan></text></g><g  id="v-409" label-idx="2" cursor="move" transform="translate(2340, 1560.6)" style="opacity: 1;"><rect id="v-411" fill="#ffffff" rx="3" ry="3" pointer-events="none"  display="none" width="0" height="0" transform="matrix(1,0,0,1,0,0)"></rect><text id="v-410" font-size="19px" y="0.8em" display="none" xml:space="preserve" text-anchor="middle" pointer-events="none" fill="#707070" font-family="Open Sans" font-weight="normal"  transform="matrix(1,0,0,1,0,0)"><tspan id="v-412"  dy="0em" x="0" style="fill-opacity: 0; stroke-opacity: 0;">-</tspan></text></g></g><g  display="none"></g><g  id="v-417" display="none"><g  id="v-395"><path  end="source" d="M 26 0 L 0 13 L 26 26 z"></path></g><g  id="v-396"><path  end="target" d="M 26 0 L 0 13 L 26 26 z"></path></g></g><g  id="v-416" display="none"><g  id="v-394" transform="translate(2430.001220703125, 1466.3076171875) "><g  event="remove"><circle r="11"></circle><path transform="scale(.8) translate(-16, -16)" d="M24.778,21.419 19.276,15.917 24.777,10.415 21.949,7.585 16.447,13.087 10.945,7.585 8.117,10.415 13.618,15.917 8.116,21.419 10.946,24.248 16.447,18.746 21.948,24.248z"></path><title>Remove link.</title></g><g  event="link:options"><circle r="11" transform="translate(25)"></circle><path fill="white" transform="scale(.55) translate(29, -16)" d="M31.229,17.736c0.064-0.571,0.104-1.148,0.104-1.736s-0.04-1.166-0.104-1.737l-4.377-1.557c-0.218-0.716-0.504-1.401-0.851-2.05l1.993-4.192c-0.725-0.91-1.549-1.734-2.458-2.459l-4.193,1.994c-0.647-0.347-1.334-0.632-2.049-0.849l-1.558-4.378C17.165,0.708,16.588,0.667,16,0.667s-1.166,0.041-1.737,0.105L12.707,5.15c-0.716,0.217-1.401,0.502-2.05,0.849L6.464,4.005C5.554,4.73,4.73,5.554,4.005,6.464l1.994,4.192c-0.347,0.648-0.632,1.334-0.849,2.05l-4.378,1.557C0.708,14.834,0.667,15.412,0.667,16s0.041,1.165,0.105,1.736l4.378,1.558c0.217,0.715,0.502,1.401,0.849,2.049l-1.994,4.193c0.725,0.909,1.549,1.733,2.459,2.458l4.192-1.993c0.648,0.347,1.334,0.633,2.05,0.851l1.557,4.377c0.571,0.064,1.148,0.104,1.737,0.104c0.588,0,1.165-0.04,1.736-0.104l1.558-4.377c0.715-0.218,1.399-0.504,2.049-0.851l4.193,1.993c0.909-0.725,1.733-1.549,2.458-2.458l-1.993-4.193c0.347-0.647,0.633-1.334,0.851-2.049L31.229,17.736zM16,20.871c-2.69,0-4.872-2.182-4.872-4.871c0-2.69,2.182-4.872,4.872-4.872c2.689,0,4.871,2.182,4.871,4.872C20.871,18.689,18.689,20.871,16,20.871z"></path><title>Link options.</title></g></g></g></g><g id="j_26"   ><path  stroke="#707070" id="v-441" stroke-width="2" stroke-dasharray="30 30" fill="none" d="M 1740 1971 1545 1941 S 1535 1940 1525 1941 L 1350 1968" style="opacity: 1;"></path><path  fill="black" stroke="black" transform="translate(1740,1971) scale(1,1) rotate(-171.40090942382812)"></path><path  fill="#707070" stroke="#707070" id="v-443" d="M 20 0 L 0 10 L 20 20 z" transform="translate(1328.5047894829493,1961.1123693136124) scale(1,1) rotate(-8.599088668823242)" style="opacity: 1;"></path><path  id="v-442" fill="none" d="M 1740 1971 1545 1941 S 1535 1940 1525 1941 L 1350 1968"></path><title ></title><g ><g  id="v-426" label-idx="0" cursor="move" transform="translate(1545.0970458984375, 1941.014892578125)" style="opacity: 1;"><rect id="v-428" fill="#ffffff" rx="3" ry="3" stroke="#ffffff" stroke-width="20px" pointer-events="none" width="156.1875" height="119" transform="matrix(1,0,0,1,-78.1,-59.5)"></rect><text id="v-427" font-size="24px" y="0.8em" xml:space="preserve" text-anchor="middle" pointer-events="none" fill="#707070" font-family="Open Sans" font-weight="normal" transform="matrix(1,0,0,1,0,-52.2)"><tspan id="v-429"  dy="0em" x="0">Sends locally</tspan><tspan id="v-430"  dy="1.2em" x="0">tuned tensors</tspan><tspan id="v-431"  dy="1.2em" x="0">and training</tspan><tspan id="v-432"  dy="1.2em" x="0">metrics</tspan></text></g><g  id="v-433" label-idx="1" cursor="move" transform="translate(1545.0970458984375, 2010.614892578125)" style="opacity: 1;"><rect id="v-435" fill="#ffffff" rx="3" ry="3" pointer-events="none" width="0" height="0" transform="matrix(1,0,0,1,0,0)"></rect><text id="v-434" font-size="19px" y="0.8em" display="none" xml:space="preserve" text-anchor="middle" pointer-events="none" fill="#707070" font-family="Open Sans" font-weight="normal" transform="matrix(1,0,0,1,0,0)"><tspan id="v-436"  dy="0em" x="0" style="fill-opacity: 0; stroke-opacity: 0;">-</tspan></text></g><g  id="v-437" label-idx="2" cursor="move" transform="translate(1545.0970458984375, 2034.614892578125)" style="opacity: 1;"><rect id="v-439" fill="#ffffff" rx="3" ry="3" pointer-events="none"  display="none" width="0" height="0" transform="matrix(1,0,0,1,0,0)"></rect><text id="v-438" font-size="19px" y="0.8em" display="none" xml:space="preserve" text-anchor="middle" pointer-events="none" fill="#707070" font-family="Open Sans" font-weight="normal"  transform="matrix(1,0,0,1,0,0)"><tspan id="v-440"  dy="0em" x="0" style="fill-opacity: 0; stroke-opacity: 0;">-</tspan></text></g></g><g  display="none"><g  transform="translate(1535, 1940)" id="v-422"><circle  idx="0" r="10"></circle><path  idx="0" d="M16,5.333c-7.732,0-14,4.701-14,10.5c0,1.982,0.741,3.833,2.016,5.414L2,25.667l5.613-1.441c2.339,1.317,5.237,2.107,8.387,2.107c7.732,0,14-4.701,14-10.5C30,10.034,23.732,5.333,16,5.333z" transform="translate(5, -33)"></path><path  idx="0" transform="scale(.8) translate(9.5, -37)" d="M24.778,21.419 19.276,15.917 24.777,10.415 21.949,7.585 16.447,13.087 10.945,7.585 8.117,10.415 13.618,15.917 8.116,21.419 10.946,24.248 16.447,18.746 21.948,24.248z"><title>Remove vertex.</title></path></g></g><g  id="v-445" display="none"><g  id="v-423"><path  end="source" d="M 26 0 L 0 13 L 26 26 z"></path></g><g  id="v-424"><path  end="target" d="M 26 0 L 0 13 L 26 26 z"></path></g></g><g  id="v-444" display="none"><g  id="v-421" transform="translate(1700.465087890625, 1964.917724609375) "><g  event="remove"><circle r="11"></circle><path transform="scale(.8) translate(-16, -16)" d="M24.778,21.419 19.276,15.917 24.777,10.415 21.949,7.585 16.447,13.087 10.945,7.585 8.117,10.415 13.618,15.917 8.116,21.419 10.946,24.248 16.447,18.746 21.948,24.248z"></path><title>Remove link.</title></g><g  event="link:options"><circle r="11" transform="translate(25)"></circle><path fill="white" transform="scale(.55) translate(29, -16)" d="M31.229,17.736c0.064-0.571,0.104-1.148,0.104-1.736s-0.04-1.166-0.104-1.737l-4.377-1.557c-0.218-0.716-0.504-1.401-0.851-2.05l1.993-4.192c-0.725-0.91-1.549-1.734-2.458-2.459l-4.193,1.994c-0.647-0.347-1.334-0.632-2.049-0.849l-1.558-4.378C17.165,0.708,16.588,0.667,16,0.667s-1.166,0.041-1.737,0.105L12.707,5.15c-0.716,0.217-1.401,0.502-2.05,0.849L6.464,4.005C5.554,4.73,4.73,5.554,4.005,6.464l1.994,4.192c-0.347,0.648-0.632,1.334-0.849,2.05l-4.378,1.557C0.708,14.834,0.667,15.412,0.667,16s0.041,1.165,0.105,1.736l4.378,1.558c0.217,0.715,0.502,1.401,0.849,2.049l-1.994,4.193c0.725,0.909,1.549,1.733,2.459,2.458l4.192-1.993c0.648,0.347,1.334,0.633,2.05,0.851l1.557,4.377c0.571,0.064,1.148,0.104,1.737,0.104c0.588,0,1.165-0.04,1.736-0.104l1.558-4.377c0.715-0.218,1.399-0.504,2.049-0.851l4.193,1.993c0.909-0.725,1.733-1.549,2.458-2.458l-1.993-4.193c0.347-0.647,0.633-1.334,0.851-2.049L31.229,17.736zM16,20.871c-2.69,0-4.872-2.182-4.872-4.871c0-2.69,2.182-4.872,4.872-4.872c2.689,0,4.871,2.182,4.871,4.872C20.871,18.689,18.689,20.871,16,20.871z"></path><title>Link options.</title></g></g></g></g><g id="j_27"   ><path  stroke="#707070" id="v-467" stroke-width="2" stroke-dasharray="30 30" fill="none" d="M 1096 600 1100 740" style="opacity: 1;"></path><path  fill="black" stroke="black" transform="translate(1096,600) scale(1,1) rotate(88.21008300781253)"></path><path  fill="#707070" stroke="#707070" id="v-469" d="M 20 0 L 0 10 L 20 20 z" transform="translate(1091.0048486091628,760.3123508866711) scale(1,1) rotate(-91.78990936279297)" style="opacity: 1;"></path><path  id="v-468" fill="none" d="M 1096 600 1100 740"></path><title ></title><g ><g  id="v-453" label-idx="0" cursor="move" transform="translate(1098, 670)" style="opacity: 1;"><rect id="v-455" fill="#ffffff" rx="3" ry="3" stroke="#ffffff" stroke-width="20px" pointer-events="none" width="199.65625" height="90.203125" transform="matrix(1,0,0,1,-99.8,-45.1)"></rect><text id="v-454" font-size="24px" y="0.8em" xml:space="preserve" text-anchor="middle" pointer-events="none" fill="#707070" font-family="Open Sans" font-weight="normal" transform="matrix(1,0,0,1,0,-37.8)"><tspan id="v-456"  dy="0em" x="0">Provides FL Plans,</tspan><tspan id="v-457"  dy="1.2em" x="0">Tasks, Models,</tspan><tspan id="v-458"  dy="1.2em" x="0">DataLoaders</tspan></text></g><g  id="v-459" label-idx="1" cursor="move" transform="translate(1098, 727.6)" style="opacity: 1;"><rect id="v-461" fill="#ffffff" rx="3" ry="3" pointer-events="none" width="0" height="0" transform="matrix(1,0,0,1,0,0)"></rect><text id="v-460" font-size="19px" y="0.8em" display="none" xml:space="preserve" text-anchor="middle" pointer-events="none" fill="#707070" font-family="Open Sans" font-weight="normal" transform="matrix(1,0,0,1,0,0)"><tspan id="v-462"  dy="0em" x="0" style="fill-opacity: 0; stroke-opacity: 0;">-</tspan></text></g><g  id="v-463" label-idx="2" cursor="move" transform="translate(1098, 751.6)" style="opacity: 1;"><rect id="v-465" fill="#ffffff" rx="3" ry="3" pointer-events="none"  display="none" width="0" height="0" transform="matrix(1,0,0,1,0,0)"></rect><text id="v-464" font-size="19px" y="0.8em" display="none" xml:space="preserve" text-anchor="middle" pointer-events="none" fill="#707070" font-family="Open Sans" font-weight="normal"  transform="matrix(1,0,0,1,0,0)"><tspan id="v-466"  dy="0em" x="0" style="fill-opacity: 0; stroke-opacity: 0;">-</tspan></text></g></g><g  display="none"></g><g  id="v-471" display="none"><g  id="v-450"><path  end="source" d="M 26 0 L 0 13 L 26 26 z"></path></g><g  id="v-451"><path  end="target" d="M 26 0 L 0 13 L 26 26 z"></path></g></g><g  id="v-470" display="none"><g  id="v-449" transform="translate(1097.142333984375, 639.9837036132812) "><g  event="remove"><circle r="11"></circle><path transform="scale(.8) translate(-16, -16)" d="M24.778,21.419 19.276,15.917 24.777,10.415 21.949,7.585 16.447,13.087 10.945,7.585 8.117,10.415 13.618,15.917 8.116,21.419 10.946,24.248 16.447,18.746 21.948,24.248z"></path><title>Remove link.</title></g><g  event="link:options"><circle r="11" transform="translate(25)"></circle><path fill="white" transform="scale(.55) translate(29, -16)" d="M31.229,17.736c0.064-0.571,0.104-1.148,0.104-1.736s-0.04-1.166-0.104-1.737l-4.377-1.557c-0.218-0.716-0.504-1.401-0.851-2.05l1.993-4.192c-0.725-0.91-1.549-1.734-2.458-2.459l-4.193,1.994c-0.647-0.347-1.334-0.632-2.049-0.849l-1.558-4.378C17.165,0.708,16.588,0.667,16,0.667s-1.166,0.041-1.737,0.105L12.707,5.15c-0.716,0.217-1.401,0.502-2.05,0.849L6.464,4.005C5.554,4.73,4.73,5.554,4.005,6.464l1.994,4.192c-0.347,0.648-0.632,1.334-0.849,2.05l-4.378,1.557C0.708,14.834,0.667,15.412,0.667,16s0.041,1.165,0.105,1.736l4.378,1.558c0.217,0.715,0.502,1.401,0.849,2.049l-1.994,4.193c0.725,0.909,1.549,1.733,2.459,2.458l4.192-1.993c0.648,0.347,1.334,0.633,2.05,0.851l1.557,4.377c0.571,0.064,1.148,0.104,1.737,0.104c0.588,0,1.165-0.04,1.736-0.104l1.558-4.377c0.715-0.218,1.399-0.504,2.049-0.851l4.193,1.993c0.909-0.725,1.733-1.549,2.458-2.458l-1.993-4.193c0.347-0.647,0.633-1.334,0.851-2.049L31.229,17.736zM16,20.871c-2.69,0-4.872-2.182-4.872-4.871c0-2.69,2.182-4.872,4.872-4.872c2.689,0,4.871,2.182,4.871,4.872C20.871,18.689,18.689,20.871,16,20.871z"></path><title>Link options.</title></g></g></g></g><g id="j_28"   ><path  stroke="#707070" id="v-493" stroke-width="2" stroke-dasharray="30 30" fill="none" d="M 1330 2047 1555 2088 S 1565 2090 1575 2088 L 1720 2057" style="opacity: 1;"></path><path  fill="black" stroke="black" transform="translate(1330,2047) scale(1,1) rotate(10.36920166015625)"></path><path  fill="#707070" stroke="#707070" id="v-495" d="M 20 0 L 0 10 L 20 20 z" transform="translate(1742.0685808356088,2062.783760202619) scale(1,1) rotate(-191.9381561279297)" style="opacity: 1;"></path><path  id="v-494" fill="none" d="M 1330 2047 1555 2088 S 1565 2090 1575 2088 L 1720 2057"></path><title ></title><g ><g  id="v-480" label-idx="0" cursor="move" transform="translate(1525.3284912109375, 2082.59326171875)" style="opacity: 1;"><rect id="v-482" fill="#ffffff" rx="3" ry="3" stroke="#ffffff" stroke-width="20px" pointer-events="none" width="180.0625" height="61.40625" transform="matrix(1,0,0,1,-90,-30.7)"></rect><text id="v-481" font-size="24px" y="0.8em" xml:space="preserve" text-anchor="middle" pointer-events="none" fill="#707070" font-family="Open Sans" font-weight="normal" transform="matrix(1,0,0,1,0,-23.4)"><tspan id="v-483"  dy="0em" x="0">Sends tasks and</tspan><tspan id="v-484"  dy="1.2em" x="0">initial tensors</tspan></text></g><g  id="v-485" label-idx="1" cursor="move" transform="translate(1525.3284912109375, 2125.79326171875)" style="opacity: 1;"><rect id="v-487" fill="#ffffff" rx="3" ry="3" pointer-events="none" width="0" height="0" transform="matrix(1,0,0,1,0,0)"></rect><text id="v-486" font-size="19px" y="0.8em" display="none" xml:space="preserve" text-anchor="middle" pointer-events="none" fill="#707070" font-family="Open Sans" font-weight="normal" transform="matrix(1,0,0,1,0,0)"><tspan id="v-488"  dy="0em" x="0" style="fill-opacity: 0; stroke-opacity: 0;">-</tspan></text></g><g  id="v-489" label-idx="2" cursor="move" transform="translate(1525.3284912109375, 2149.79326171875)" style="opacity: 1;"><rect id="v-491" fill="#ffffff" rx="3" ry="3" pointer-events="none"  display="none" width="0" height="0" transform="matrix(1,0,0,1,0,0)"></rect><text id="v-490" font-size="19px" y="0.8em" display="none" xml:space="preserve" text-anchor="middle" pointer-events="none" fill="#707070" font-family="Open Sans" font-weight="normal"  transform="matrix(1,0,0,1,0,0)"><tspan id="v-492"  dy="0em" x="0" style="fill-opacity: 0; stroke-opacity: 0;">-</tspan></text></g></g><g  display="none"><g  transform="translate(1565, 2090)" id="v-476"><circle  idx="0" r="10"></circle><path  idx="0" d="M16,5.333c-7.732,0-14,4.701-14,10.5c0,1.982,0.741,3.833,2.016,5.414L2,25.667l5.613-1.441c2.339,1.317,5.237,2.107,8.387,2.107c7.732,0,14-4.701,14-10.5C30,10.034,23.732,5.333,16,5.333z" transform="translate(5, -33)"></path><path  idx="0" transform="scale(.8) translate(9.5, -37)" d="M24.778,21.419 19.276,15.917 24.777,10.415 21.949,7.585 16.447,13.087 10.945,7.585 8.117,10.415 13.618,15.917 8.116,21.419 10.946,24.248 16.447,18.746 21.948,24.248z"><title>Remove vertex.</title></path></g></g><g  id="v-497" display="none"><g  id="v-477"><path  end="source" d="M 26 0 L 0 13 L 26 26 z"></path></g><g  id="v-478"><path  end="target" d="M 26 0 L 0 13 L 26 26 z"></path></g></g><g  id="v-496" display="none"><g  id="v-475" transform="translate(1369.35205078125, 2054.1708984375) "><g  event="remove"><circle r="11"></circle><path transform="scale(.8) translate(-16, -16)" d="M24.778,21.419 19.276,15.917 24.777,10.415 21.949,7.585 16.447,13.087 10.945,7.585 8.117,10.415 13.618,15.917 8.116,21.419 10.946,24.248 16.447,18.746 21.948,24.248z"></path><title>Remove link.</title></g><g  event="link:options"><circle r="11" transform="translate(25)"></circle><path fill="white" transform="scale(.55) translate(29, -16)" d="M31.229,17.736c0.064-0.571,0.104-1.148,0.104-1.736s-0.04-1.166-0.104-1.737l-4.377-1.557c-0.218-0.716-0.504-1.401-0.851-2.05l1.993-4.192c-0.725-0.91-1.549-1.734-2.458-2.459l-4.193,1.994c-0.647-0.347-1.334-0.632-2.049-0.849l-1.558-4.378C17.165,0.708,16.588,0.667,16,0.667s-1.166,0.041-1.737,0.105L12.707,5.15c-0.716,0.217-1.401,0.502-2.05,0.849L6.464,4.005C5.554,4.73,4.73,5.554,4.005,6.464l1.994,4.192c-0.347,0.648-0.632,1.334-0.849,2.05l-4.378,1.557C0.708,14.834,0.667,15.412,0.667,16s0.041,1.165,0.105,1.736l4.378,1.558c0.217,0.715,0.502,1.401,0.849,2.049l-1.994,4.193c0.725,0.909,1.549,1.733,2.459,2.458l4.192-1.993c0.648,0.347,1.334,0.633,2.05,0.851l1.557,4.377c0.571,0.064,1.148,0.104,1.737,0.104c0.588,0,1.165-0.04,1.736-0.104l1.558-4.377c0.715-0.218,1.399-0.504,2.049-0.851l4.193,1.993c0.909-0.725,1.733-1.549,2.458-2.458l-1.993-4.193c0.347-0.647,0.633-1.334,0.851-2.049L31.229,17.736zM16,20.871c-2.69,0-4.872-2.182-4.872-4.871c0-2.69,2.182-4.872,4.872-4.872c2.689,0,4.871,2.182,4.871,4.872C20.871,18.689,18.689,20.871,16,20.871z"></path><title>Link options.</title></g></g></g></g><g id="j_29"   ><path  stroke="#707070" id="v-519" stroke-width="2" stroke-dasharray="30 30" fill="none" d="M 1330 1500 1540 1529 S 1550 1530 1560 1529 L 1720 1506" style="opacity: 1;"></path><path  fill="black" stroke="black" transform="translate(1330,1500) scale(1,1) rotate(7.765167236328139)"></path><path  fill="#707070" stroke="#707070" id="v-521" d="M 20 0 L 0 10 L 20 20 z" transform="translate(1741.4069390794684,1512.9004776234322) scale(1,1) rotate(-188.08787536621094)" style="opacity: 1;"></path><path  id="v-520" fill="none" d="M 1330 1500 1540 1529 S 1550 1530 1560 1529 L 1720 1506"></path><title ></title><g ><g  id="v-506" label-idx="0" cursor="move" transform="translate(1524.98193359375, 1526.926025390625)" style="opacity: 1;"><rect id="v-508" fill="#ffffff" rx="3" ry="3" stroke="#ffffff" stroke-width="20px" pointer-events="none" width="185.4375" height="61.40625" transform="matrix(1,0,0,1,-92.7,-30.7)"></rect><text id="v-507" font-size="24px" y="0.8em" xml:space="preserve" text-anchor="middle" pointer-events="none" fill="#707070" font-family="Open Sans" font-weight="normal" transform="matrix(1,0,0,1,0,-23.4)"><tspan id="v-509"  dy="0em" x="0">Approves, Sends</tspan><tspan id="v-510"  dy="1.2em" x="0">FL experiments</tspan></text></g><g  id="v-511" label-idx="1" cursor="move" transform="translate(1524.98193359375, 1570.126025390625)" style="opacity: 1;"><rect id="v-513" fill="#ffffff" rx="3" ry="3" pointer-events="none" width="0" height="0" transform="matrix(1,0,0,1,0,0)"></rect><text id="v-512" font-size="19px" y="0.8em" display="none" xml:space="preserve" text-anchor="middle" pointer-events="none" fill="#707070" font-family="Open Sans" font-weight="normal" transform="matrix(1,0,0,1,0,0)"><tspan id="v-514"  dy="0em" x="0" style="fill-opacity: 0; stroke-opacity: 0;">-</tspan></text></g><g  id="v-515" label-idx="2" cursor="move" transform="translate(1524.98193359375, 1594.126025390625)" style="opacity: 1;"><rect id="v-517" fill="#ffffff" rx="3" ry="3" pointer-events="none"  display="none" width="0" height="0" transform="matrix(1,0,0,1,0,0)"></rect><text id="v-516" font-size="19px" y="0.8em" display="none" xml:space="preserve" text-anchor="middle" pointer-events="none" fill="#707070" font-family="Open Sans" font-weight="normal"  transform="matrix(1,0,0,1,0,0)"><tspan id="v-518"  dy="0em" x="0" style="fill-opacity: 0; stroke-opacity: 0;">-</tspan></text></g></g><g  display="none"><g  transform="translate(1550, 1530)" id="v-502"><circle  idx="0" r="10"></circle><path  idx="0" d="M16,5.333c-7.732,0-14,4.701-14,10.5c0,1.982,0.741,3.833,2.016,5.414L2,25.667l5.613-1.441c2.339,1.317,5.237,2.107,8.387,2.107c7.732,0,14-4.701,14-10.5C30,10.034,23.732,5.333,16,5.333z" transform="translate(5, -33)"></path><path  idx="0" transform="scale(.8) translate(9.5, -37)" d="M24.778,21.419 19.276,15.917 24.777,10.415 21.949,7.585 16.447,13.087 10.945,7.585 8.117,10.415 13.618,15.917 8.116,21.419 10.946,24.248 16.447,18.746 21.948,24.248z"><title>Remove vertex.</title></path></g></g><g  id="v-523" display="none"><g  id="v-503"><path  end="source" d="M 26 0 L 0 13 L 26 26 z"></path></g><g  id="v-504"><path  end="target" d="M 26 0 L 0 13 L 26 26 z"></path></g></g><g  id="v-522" display="none"><g  id="v-501" transform="translate(1369.6240234375, 1505.471923828125) "><g  event="remove"><circle r="11"></circle><path transform="scale(.8) translate(-16, -16)" d="M24.778,21.419 19.276,15.917 24.777,10.415 21.949,7.585 16.447,13.087 10.945,7.585 8.117,10.415 13.618,15.917 8.116,21.419 10.946,24.248 16.447,18.746 21.948,24.248z"></path><title>Remove link.</title></g><g  event="link:options"><circle r="11" transform="translate(25)"></circle><path fill="white" transform="scale(.55) translate(29, -16)" d="M31.229,17.736c0.064-0.571,0.104-1.148,0.104-1.736s-0.04-1.166-0.104-1.737l-4.377-1.557c-0.218-0.716-0.504-1.401-0.851-2.05l1.993-4.192c-0.725-0.91-1.549-1.734-2.458-2.459l-4.193,1.994c-0.647-0.347-1.334-0.632-2.049-0.849l-1.558-4.378C17.165,0.708,16.588,0.667,16,0.667s-1.166,0.041-1.737,0.105L12.707,5.15c-0.716,0.217-1.401,0.502-2.05,0.849L6.464,4.005C5.554,4.73,4.73,5.554,4.005,6.464l1.994,4.192c-0.347,0.648-0.632,1.334-0.849,2.05l-4.378,1.557C0.708,14.834,0.667,15.412,0.667,16s0.041,1.165,0.105,1.736l4.378,1.558c0.217,0.715,0.502,1.401,0.849,2.049l-1.994,4.193c0.725,0.909,1.549,1.733,2.459,2.458l4.192-1.993c0.648,0.347,1.334,0.633,2.05,0.851l1.557,4.377c0.571,0.064,1.148,0.104,1.737,0.104c0.588,0,1.165-0.04,1.736-0.104l1.558-4.377c0.715-0.218,1.399-0.504,2.049-0.851l4.193,1.993c0.909-0.725,1.733-1.549,2.458-2.458l-1.993-4.193c0.347-0.647,0.633-1.334,0.851-2.049L31.229,17.736zM16,20.871c-2.69,0-4.872-2.182-4.872-4.871c0-2.69,2.182-4.872,4.872-4.872c2.689,0,4.871,2.182,4.871,4.872C20.871,18.689,18.689,20.871,16,20.871z"></path><title>Link options.</title></g></g></g></g><g id="j_30"   ><path  stroke="#707070" id="v-547" stroke-width="2" stroke-dasharray="30 30" fill="none" d="M 1740 1413 1540 1362 S 1530 1360 1520 1363 L 1349 1407" style="opacity: 1;"></path><path  fill="black" stroke="black" transform="translate(1740,1413) scale(1,1) rotate(-165.83543395996094)"></path><path  fill="#707070" stroke="#707070" id="v-549" d="M 20 0 L 0 10 L 20 20 z" transform="translate(1327.4836591093501,1402.3217559621125) scale(1,1) rotate(-14.57421588897705)" style="opacity: 1;"></path><path  id="v-548" fill="none" d="M 1740 1413 1540 1362 S 1530 1360 1520 1363 L 1349 1407"></path><title ></title><g ><g  id="v-532" label-idx="0" cursor="move" transform="translate(1544.66943359375, 1363.190673828125)" style="opacity: 1;"><rect id="v-534" fill="#ffffff" rx="3" ry="3" stroke="#ffffff" stroke-width="20px" pointer-events="none" width="167.9375" height="119" transform="matrix(1,0,0,1,-84,-59.5)"></rect><text id="v-533" font-size="24px" y="0.8em" xml:space="preserve" text-anchor="middle" pointer-events="none" fill="#707070" font-family="Open Sans" font-weight="normal" transform="matrix(1,0,0,1,0,-52.2)"><tspan id="v-535"  dy="0em" x="0">Communicates</tspan><tspan id="v-536"  dy="1.2em" x="0">dataset info,</tspan><tspan id="v-537"  dy="1.2em" x="0">Sends status</tspan><tspan id="v-538"  dy="1.2em" x="0">updates</tspan></text></g><g  id="v-539" label-idx="1" cursor="move" transform="translate(1544.66943359375, 1432.790673828125)" style="opacity: 1;"><rect id="v-541" fill="#ffffff" rx="3" ry="3" pointer-events="none" width="0" height="0" transform="matrix(1,0,0,1,0,0)"></rect><text id="v-540" font-size="19px" y="0.8em" display="none" xml:space="preserve" text-anchor="middle" pointer-events="none" fill="#707070" font-family="Open Sans" font-weight="normal" transform="matrix(1,0,0,1,0,0)"><tspan id="v-542"  dy="0em" x="0" style="fill-opacity: 0; stroke-opacity: 0;">-</tspan></text></g><g  id="v-543" label-idx="2" cursor="move" transform="translate(1544.66943359375, 1456.790673828125)" style="opacity: 1;"><rect id="v-545" fill="#ffffff" rx="3" ry="3" pointer-events="none"  display="none" width="0" height="0" transform="matrix(1,0,0,1,0,0)"></rect><text id="v-544" font-size="19px" y="0.8em" display="none" xml:space="preserve" text-anchor="middle" pointer-events="none" fill="#707070" font-family="Open Sans" font-weight="normal"  transform="matrix(1,0,0,1,0,0)"><tspan id="v-546"  dy="0em" x="0" style="fill-opacity: 0; stroke-opacity: 0;">-</tspan></text></g></g><g  display="none"><g  transform="translate(1530, 1360)" id="v-528"><circle  idx="0" r="10"></circle><path  idx="0" d="M16,5.333c-7.732,0-14,4.701-14,10.5c0,1.982,0.741,3.833,2.016,5.414L2,25.667l5.613-1.441c2.339,1.317,5.237,2.107,8.387,2.107c7.732,0,14-4.701,14-10.5C30,10.034,23.732,5.333,16,5.333z" transform="translate(5, -33)"></path><path  idx="0" transform="scale(.8) translate(9.5, -37)" d="M24.778,21.419 19.276,15.917 24.777,10.415 21.949,7.585 16.447,13.087 10.945,7.585 8.117,10.415 13.618,15.917 8.116,21.419 10.946,24.248 16.447,18.746 21.948,24.248z"><title>Remove vertex.</title></path></g></g><g  id="v-551" display="none"><g  id="v-529"><path  end="source" d="M 26 0 L 0 13 L 26 26 z"></path></g><g  id="v-530"><path  end="target" d="M 26 0 L 0 13 L 26 26 z"></path></g></g><g  id="v-550" display="none"><g  id="v-527" transform="translate(1701.2403564453125, 1403.1163330078125) "><g  event="remove"><circle r="11"></circle><path transform="scale(.8) translate(-16, -16)" d="M24.778,21.419 19.276,15.917 24.777,10.415 21.949,7.585 16.447,13.087 10.945,7.585 8.117,10.415 13.618,15.917 8.116,21.419 10.946,24.248 16.447,18.746 21.948,24.248z"></path><title>Remove link.</title></g><g  event="link:options"><circle r="11" transform="translate(25)"></circle><path fill="white" transform="scale(.55) translate(29, -16)" d="M31.229,17.736c0.064-0.571,0.104-1.148,0.104-1.736s-0.04-1.166-0.104-1.737l-4.377-1.557c-0.218-0.716-0.504-1.401-0.851-2.05l1.993-4.192c-0.725-0.91-1.549-1.734-2.458-2.459l-4.193,1.994c-0.647-0.347-1.334-0.632-2.049-0.849l-1.558-4.378C17.165,0.708,16.588,0.667,16,0.667s-1.166,0.041-1.737,0.105L12.707,5.15c-0.716,0.217-1.401,0.502-2.05,0.849L6.464,4.005C5.554,4.73,4.73,5.554,4.005,6.464l1.994,4.192c-0.347,0.648-0.632,1.334-0.849,2.05l-4.378,1.557C0.708,14.834,0.667,15.412,0.667,16s0.041,1.165,0.105,1.736l4.378,1.558c0.217,0.715,0.502,1.401,0.849,2.049l-1.994,4.193c0.725,0.909,1.549,1.733,2.459,2.458l4.192-1.993c0.648,0.347,1.334,0.633,2.05,0.851l1.557,4.377c0.571,0.064,1.148,0.104,1.737,0.104c0.588,0,1.165-0.04,1.736-0.104l1.558-4.377c0.715-0.218,1.399-0.504,2.049-0.851l4.193,1.993c0.909-0.725,1.733-1.549,2.458-2.458l-1.993-4.193c0.347-0.647,0.633-1.334,0.851-2.049L31.229,17.736zM16,20.871c-2.69,0-4.872-2.182-4.872-4.871c0-2.69,2.182-4.872,4.872-4.872c2.689,0,4.871,2.182,4.871,4.872C20.871,18.689,18.689,20.871,16,20.871z"></path><title>Link options.</title></g></g></g></g><g id="j_31"   ><path  stroke="#707070" id="v-574" stroke-width="2" stroke-dasharray="30 30" fill="none" d="M 1965 1620 1965 1835" style="opacity: 1;"></path><path  fill="black" stroke="black" transform="translate(1965,1620) scale(1,1) rotate(-270)"></path><path  fill="#707070" stroke="#707070" id="v-576" d="M 20 0 L 0 10 L 20 20 z" transform="translate(1955,1855) scale(1,1) rotate(-90)" style="opacity: 1;"></path><path  id="v-575" fill="none" d="M 1965 1620 1965 1835"></path><title ></title><g ><g  id="v-559" label-idx="0" cursor="move" transform="translate(1965, 1727.5)" style="opacity: 1;"><rect id="v-561" fill="#ffffff" rx="3" ry="3" stroke="#ffffff" stroke-width="20px" pointer-events="none" width="164.46875" height="119" transform="matrix(1,0,0,1,-82.2,-59.5)"></rect><text id="v-560" font-size="24px" y="0.8em" xml:space="preserve" text-anchor="middle" pointer-events="none" fill="#707070" font-family="Open Sans" font-weight="normal" transform="matrix(1,0,0,1,0,-52.2)"><tspan id="v-562"  dy="0em" x="0">Creates an</tspan><tspan id="v-563"  dy="1.2em" x="0">instance to</tspan><tspan id="v-564"  dy="1.2em" x="0">maintain an FL</tspan><tspan id="v-565"  dy="1.2em" x="0">experiment</tspan></text></g><g  id="v-566" label-idx="1" cursor="move" transform="translate(1965, 1797.1)" style="opacity: 1;"><rect id="v-568" fill="#ffffff" rx="3" ry="3" pointer-events="none" width="0" height="0" transform="matrix(1,0,0,1,0,0)"></rect><text id="v-567" font-size="19px" y="0.8em" display="none" xml:space="preserve" text-anchor="middle" pointer-events="none" fill="#707070" font-family="Open Sans" font-weight="normal" transform="matrix(1,0,0,1,0,0)"><tspan id="v-569"  dy="0em" x="0" style="fill-opacity: 0; stroke-opacity: 0;">-</tspan></text></g><g  id="v-570" label-idx="2" cursor="move" transform="translate(1965, 1821.1)" style="opacity: 1;"><rect id="v-572" fill="#ffffff" rx="3" ry="3" pointer-events="none"  display="none" width="0" height="0" transform="matrix(1,0,0,1,0,0)"></rect><text id="v-571" font-size="19px" y="0.8em" display="none" xml:space="preserve" text-anchor="middle" pointer-events="none" fill="#707070" font-family="Open Sans" font-weight="normal"  transform="matrix(1,0,0,1,0,0)"><tspan id="v-573"  dy="0em" x="0" style="fill-opacity: 0; stroke-opacity: 0;">-</tspan></text></g></g><g  display="none"></g><g  id="v-578" display="none"><g  id="v-556"><path  end="source" d="M 26 0 L 0 13 L 26 26 z"></path></g><g  id="v-557"><path  end="target" d="M 26 0 L 0 13 L 26 26 z"></path></g></g><g  id="v-577" display="none"><g  id="v-555" transform="translate(1965, 1660) "><g  event="remove"><circle r="11"></circle><path transform="scale(.8) translate(-16, -16)" d="M24.778,21.419 19.276,15.917 24.777,10.415 21.949,7.585 16.447,13.087 10.945,7.585 8.117,10.415 13.618,15.917 8.116,21.419 10.946,24.248 16.447,18.746 21.948,24.248z"></path><title>Remove link.</title></g><g  event="link:options"><circle r="11" transform="translate(25)"></circle><path fill="white" transform="scale(.55) translate(29, -16)" d="M31.229,17.736c0.064-0.571,0.104-1.148,0.104-1.736s-0.04-1.166-0.104-1.737l-4.377-1.557c-0.218-0.716-0.504-1.401-0.851-2.05l1.993-4.192c-0.725-0.91-1.549-1.734-2.458-2.459l-4.193,1.994c-0.647-0.347-1.334-0.632-2.049-0.849l-1.558-4.378C17.165,0.708,16.588,0.667,16,0.667s-1.166,0.041-1.737,0.105L12.707,5.15c-0.716,0.217-1.401,0.502-2.05,0.849L6.464,4.005C5.554,4.73,4.73,5.554,4.005,6.464l1.994,4.192c-0.347,0.648-0.632,1.334-0.849,2.05l-4.378,1.557C0.708,14.834,0.667,15.412,0.667,16s0.041,1.165,0.105,1.736l4.378,1.558c0.217,0.715,0.502,1.401,0.849,2.049l-1.994,4.193c0.725,0.909,1.549,1.733,2.459,2.458l4.192-1.993c0.648,0.347,1.334,0.633,2.05,0.851l1.557,4.377c0.571,0.064,1.148,0.104,1.737,0.104c0.588,0,1.165-0.04,1.736-0.104l1.558-4.377c0.715-0.218,1.399-0.504,2.049-0.851l4.193,1.993c0.909-0.725,1.733-1.549,2.458-2.458l-1.993-4.193c0.347-0.647,0.633-1.334,0.851-2.049L31.229,17.736zM16,20.871c-2.69,0-4.872-2.182-4.872-4.871c0-2.69,2.182-4.872,4.872-4.872c2.689,0,4.871,2.182,4.871,4.872C20.871,18.689,18.689,20.871,16,20.871z"></path><title>Link options.</title></g></g></g></g><g id="j_32"   ><path  stroke="#707070" id="v-601" stroke-width="2" stroke-dasharray="30 30" fill="none" d="M 1105 1620 1105 1835" style="opacity: 1;"></path><path  fill="black" stroke="black" transform="translate(1105,1620) scale(1,1) rotate(-270)"></path><path  fill="#707070" stroke="#707070" id="v-603" d="M 20 0 L 0 10 L 20 20 z" transform="translate(1095,1855) scale(1,1) rotate(-90)" style="opacity: 1;"></path><path  id="v-602" fill="none" d="M 1105 1620 1105 1835"></path><title ></title><g ><g  id="v-586" label-idx="0" cursor="move" transform="translate(1105, 1727.5)" style="opacity: 1;"><rect id="v-588" fill="#ffffff" rx="3" ry="3" stroke="#ffffff" stroke-width="20px" pointer-events="none" width="164.46875" height="119" transform="matrix(1,0,0,1,-82.2,-59.5)"></rect><text id="v-587" font-size="24px" y="0.8em" xml:space="preserve" text-anchor="middle" pointer-events="none" fill="#707070" font-family="Open Sans" font-weight="normal" transform="matrix(1,0,0,1,0,-52.2)"><tspan id="v-589"  dy="0em" x="0">Creates an</tspan><tspan id="v-590"  dy="1.2em" x="0">instance to</tspan><tspan id="v-591"  dy="1.2em" x="0">maintain an FL</tspan><tspan id="v-592"  dy="1.2em" x="0">experiment</tspan></text></g><g  id="v-593" label-idx="1" cursor="move" transform="translate(1105, 1797.1)" style="opacity: 1;"><rect id="v-595" fill="#ffffff" rx="3" ry="3" pointer-events="none" width="0" height="0" transform="matrix(1,0,0,1,0,0)"></rect><text id="v-594" font-size="19px" y="0.8em" display="none" xml:space="preserve" text-anchor="middle" pointer-events="none" fill="#707070" font-family="Open Sans" font-weight="normal" transform="matrix(1,0,0,1,0,0)"><tspan id="v-596"  dy="0em" x="0" style="fill-opacity: 0; stroke-opacity: 0;">-</tspan></text></g><g  id="v-597" label-idx="2" cursor="move" transform="translate(1105, 1821.1)" style="opacity: 1;"><rect id="v-599" fill="#ffffff" rx="3" ry="3" pointer-events="none"  display="none" width="0" height="0" transform="matrix(1,0,0,1,0,0)"></rect><text id="v-598" font-size="19px" y="0.8em" display="none" xml:space="preserve" text-anchor="middle" pointer-events="none" fill="#707070" font-family="Open Sans" font-weight="normal"  transform="matrix(1,0,0,1,0,0)"><tspan id="v-600"  dy="0em" x="0" style="fill-opacity: 0; stroke-opacity: 0;">-</tspan></text></g></g><g  display="none"></g><g  id="v-605" display="none"><g  id="v-583"><path  end="source" d="M 26 0 L 0 13 L 26 26 z"></path></g><g  id="v-584"><path  end="target" d="M 26 0 L 0 13 L 26 26 z"></path></g></g><g  id="v-604" display="none"><g  id="v-582" transform="translate(1105, 1660) "><g  event="remove"><circle r="11"></circle><path transform="scale(.8) translate(-16, -16)" d="M24.778,21.419 19.276,15.917 24.777,10.415 21.949,7.585 16.447,13.087 10.945,7.585 8.117,10.415 13.618,15.917 8.116,21.419 10.946,24.248 16.447,18.746 21.948,24.248z"></path><title>Remove link.</title></g><g  event="link:options"><circle r="11" transform="translate(25)"></circle><path fill="white" transform="scale(.55) translate(29, -16)" d="M31.229,17.736c0.064-0.571,0.104-1.148,0.104-1.736s-0.04-1.166-0.104-1.737l-4.377-1.557c-0.218-0.716-0.504-1.401-0.851-2.05l1.993-4.192c-0.725-0.91-1.549-1.734-2.458-2.459l-4.193,1.994c-0.647-0.347-1.334-0.632-2.049-0.849l-1.558-4.378C17.165,0.708,16.588,0.667,16,0.667s-1.166,0.041-1.737,0.105L12.707,5.15c-0.716,0.217-1.401,0.502-2.05,0.849L6.464,4.005C5.554,4.73,4.73,5.554,4.005,6.464l1.994,4.192c-0.347,0.648-0.632,1.334-0.849,2.05l-4.378,1.557C0.708,14.834,0.667,15.412,0.667,16s0.041,1.165,0.105,1.736l4.378,1.558c0.217,0.715,0.502,1.401,0.849,2.049l-1.994,4.193c0.725,0.909,1.549,1.733,2.459,2.458l4.192-1.993c0.648,0.347,1.334,0.633,2.05,0.851l1.557,4.377c0.571,0.064,1.148,0.104,1.737,0.104c0.588,0,1.165-0.04,1.736-0.104l1.558-4.377c0.715-0.218,1.399-0.504,2.049-0.851l4.193,1.993c0.909-0.725,1.733-1.549,2.458-2.458l-1.993-4.193c0.347-0.647,0.633-1.334,0.851-2.049L31.229,17.736zM16,20.871c-2.69,0-4.872-2.182-4.872-4.871c0-2.69,2.182-4.872,4.872-4.872c2.689,0,4.871,2.182,4.871,4.872C20.871,18.689,18.689,20.871,16,20.871z"></path><title>Link options.</title></g></g></g></g><g id="j_33"   ><path  stroke="#707070" id="v-631" stroke-width="2" stroke-dasharray="30 30" fill="none" d="M 1163 1320 1211 1194 S 1215 1185 1211 1176 L 1172 1079" style="opacity: 1;"></path><path  fill="black" stroke="black" transform="translate(1163,1320) scale(1,1) rotate(-68.93405151367188)"></path><path  fill="#707070" stroke="#707070" id="v-633" d="M 20 0 L 0 10 L 20 20 z" transform="translate(1174.284724466815,1056.286115138385) scale(1,1) rotate(68.1985778808594)" style="opacity: 1;"></path><path  id="v-632" fill="none" d="M 1163 1320 1211 1194 S 1215 1185 1211 1176 L 1172 1079"></path><title ></title><g ><g  id="v-614" label-idx="0" cursor="move" transform="translate(1208.896728515625, 1199.5211181640625)" style="opacity: 1;"><rect id="v-616" fill="#ffffff" rx="3" ry="3" stroke="#ffffff" stroke-width="20px" pointer-events="none" width="180.0625" height="176.59375" transform="matrix(1,0,0,1,-90,-88.3)"></rect><text id="v-615" font-size="24px" y="0.8em" xml:space="preserve" text-anchor="middle" pointer-events="none" fill="#707070" font-family="Open Sans" font-weight="normal" transform="matrix(1,0,0,1,0,-81)"><tspan id="v-617"  dy="0em" x="0">Sends</tspan><tspan id="v-618"  dy="1.2em" x="0">information</tspan><tspan id="v-619"  dy="1.2em" x="0">about the</tspan><tspan id="v-620"  dy="1.2em" x="0">Federation.</tspan><tspan id="v-621"  dy="1.2em" x="0">Returns training</tspan><tspan id="v-622"  dy="1.2em" x="0">artifacts.</tspan></text></g><g  id="v-623" label-idx="1" cursor="move" transform="translate(1208.896728515625, 1293.1211181640624)" style="opacity: 1;"><rect id="v-625" fill="#ffffff" rx="3" ry="3" pointer-events="none" width="0" height="0" transform="matrix(1,0,0,1,0,0)"></rect><text id="v-624" font-size="19px" y="0.8em" display="none" xml:space="preserve" text-anchor="middle" pointer-events="none" fill="#707070" font-family="Open Sans" font-weight="normal" transform="matrix(1,0,0,1,0,0)"><tspan id="v-626"  dy="0em" x="0" style="fill-opacity: 0; stroke-opacity: 0;">-</tspan></text></g><g  id="v-627" label-idx="2" cursor="move" transform="translate(1208.896728515625, 1317.1211181640624)" style="opacity: 1;"><rect id="v-629" fill="#ffffff" rx="3" ry="3" pointer-events="none"  display="none" width="0" height="0" transform="matrix(1,0,0,1,0,0)"></rect><text id="v-628" font-size="19px" y="0.8em" display="none" xml:space="preserve" text-anchor="middle" pointer-events="none" fill="#707070" font-family="Open Sans" font-weight="normal"  transform="matrix(1,0,0,1,0,0)"><tspan id="v-630"  dy="0em" x="0" style="fill-opacity: 0; stroke-opacity: 0;">-</tspan></text></g></g><g  display="none"><g  transform="translate(1215, 1185)" id="v-610"><circle  idx="0" r="10"></circle><path  idx="0" d="M16,5.333c-7.732,0-14,4.701-14,10.5c0,1.982,0.741,3.833,2.016,5.414L2,25.667l5.613-1.441c2.339,1.317,5.237,2.107,8.387,2.107c7.732,0,14-4.701,14-10.5C30,10.034,23.732,5.333,16,5.333z" transform="translate(5, -33)"></path><path  idx="0" transform="scale(.8) translate(9.5, -37)" d="M24.778,21.419 19.276,15.917 24.777,10.415 21.949,7.585 16.447,13.087 10.945,7.585 8.117,10.415 13.618,15.917 8.116,21.419 10.946,24.248 16.447,18.746 21.948,24.248z"><title>Remove vertex.</title></path></g></g><g  id="v-635" display="none"><g  id="v-611"><path  end="source" d="M 26 0 L 0 13 L 26 26 z"></path></g><g  id="v-612"><path  end="target" d="M 26 0 L 0 13 L 26 26 z"></path></g></g><g  id="v-634" display="none"><g  id="v-609" transform="translate(1177.2398681640625, 1282.6204833984375) "><g  event="remove"><circle r="11"></circle><path transform="scale(.8) translate(-16, -16)" d="M24.778,21.419 19.276,15.917 24.777,10.415 21.949,7.585 16.447,13.087 10.945,7.585 8.117,10.415 13.618,15.917 8.116,21.419 10.946,24.248 16.447,18.746 21.948,24.248z"></path><title>Remove link.</title></g><g  event="link:options"><circle r="11" transform="translate(25)"></circle><path fill="white" transform="scale(.55) translate(29, -16)" d="M31.229,17.736c0.064-0.571,0.104-1.148,0.104-1.736s-0.04-1.166-0.104-1.737l-4.377-1.557c-0.218-0.716-0.504-1.401-0.851-2.05l1.993-4.192c-0.725-0.91-1.549-1.734-2.458-2.459l-4.193,1.994c-0.647-0.347-1.334-0.632-2.049-0.849l-1.558-4.378C17.165,0.708,16.588,0.667,16,0.667s-1.166,0.041-1.737,0.105L12.707,5.15c-0.716,0.217-1.401,0.502-2.05,0.849L6.464,4.005C5.554,4.73,4.73,5.554,4.005,6.464l1.994,4.192c-0.347,0.648-0.632,1.334-0.849,2.05l-4.378,1.557C0.708,14.834,0.667,15.412,0.667,16s0.041,1.165,0.105,1.736l4.378,1.558c0.217,0.715,0.502,1.401,0.849,2.049l-1.994,4.193c0.725,0.909,1.549,1.733,2.459,2.458l4.192-1.993c0.648,0.347,1.334,0.633,2.05,0.851l1.557,4.377c0.571,0.064,1.148,0.104,1.737,0.104c0.588,0,1.165-0.04,1.736-0.104l1.558-4.377c0.715-0.218,1.399-0.504,2.049-0.851l4.193,1.993c0.909-0.725,1.733-1.549,2.458-2.458l-1.993-4.193c0.347-0.647,0.633-1.334,0.851-2.049L31.229,17.736zM16,20.871c-2.69,0-4.872-2.182-4.872-4.871c0-2.69,2.182-4.872,4.872-4.872c2.689,0,4.871,2.182,4.871,4.872C20.871,18.689,18.689,20.871,16,20.871z"></path><title>Link options.</title></g></g></g></g><g id="j_34"   ><path  stroke="#707070" id="v-657" stroke-width="2" stroke-dasharray="30 30" fill="none" d="M 1043 1060 999 1166 S 995 1175 998 1184 L 1042 1301" style="opacity: 1;"></path><path  fill="black" stroke="black" transform="translate(1043,1060) scale(1,1) rotate(-247.34477233886722)"></path><path  fill="#707070" stroke="#707070" id="v-659" d="M 20 0 L 0 10 L 20 20 z" transform="translate(1039.628751669837,1323.4899973110653) scale(1,1) rotate(-110.426025390625)" style="opacity: 1;"></path><path  id="v-658" fill="none" d="M 1043 1060 999 1166 S 995 1175 998 1184 L 1042 1301"></path><title ></title><g ><g  id="v-644" label-idx="0" cursor="move" transform="translate(997.0918579101562, 1180.0296630859375)" style="opacity: 1;"><rect id="v-646" fill="#ffffff" rx="3" ry="3" stroke="#ffffff" stroke-width="20px" pointer-events="none" width="140.5" height="61.40625" transform="matrix(1,0,0,1,-70.2,-30.7)"></rect><text id="v-645" font-size="24px" y="0.8em" xml:space="preserve" text-anchor="middle" pointer-events="none" fill="#707070" font-family="Open Sans" font-weight="normal" transform="matrix(1,0,0,1,0,-23.4)"><tspan id="v-647"  dy="0em" x="0">Registers FL</tspan><tspan id="v-648"  dy="1.2em" x="0">experiments</tspan></text></g><g  id="v-649" label-idx="1" cursor="move" transform="translate(997.0918579101562, 1223.2296630859375)" style="opacity: 1;"><rect id="v-651" fill="#ffffff" rx="3" ry="3" pointer-events="none" width="0" height="0" transform="matrix(1,0,0,1,0,0)"></rect><text id="v-650" font-size="19px" y="0.8em" display="none" xml:space="preserve" text-anchor="middle" pointer-events="none" fill="#707070" font-family="Open Sans" font-weight="normal" transform="matrix(1,0,0,1,0,0)"><tspan id="v-652"  dy="0em" x="0" style="fill-opacity: 0; stroke-opacity: 0;">-</tspan></text></g><g  id="v-653" label-idx="2" cursor="move" transform="translate(997.0918579101562, 1247.2296630859375)" style="opacity: 1;"><rect id="v-655" fill="#ffffff" rx="3" ry="3" pointer-events="none"  display="none" width="0" height="0" transform="matrix(1,0,0,1,0,0)"></rect><text id="v-654" font-size="19px" y="0.8em" display="none" xml:space="preserve" text-anchor="middle" pointer-events="none" fill="#707070" font-family="Open Sans" font-weight="normal"  transform="matrix(1,0,0,1,0,0)"><tspan id="v-656"  dy="0em" x="0" style="fill-opacity: 0; stroke-opacity: 0;">-</tspan></text></g></g><g  display="none"><g  transform="translate(995, 1175)" id="v-640"><circle  idx="0" r="10"></circle><path  idx="0" d="M16,5.333c-7.732,0-14,4.701-14,10.5c0,1.982,0.741,3.833,2.016,5.414L2,25.667l5.613-1.441c2.339,1.317,5.237,2.107,8.387,2.107c7.732,0,14-4.701,14-10.5C30,10.034,23.732,5.333,16,5.333z" transform="translate(5, -33)"></path><path  idx="0" transform="scale(.8) translate(9.5, -37)" d="M24.778,21.419 19.276,15.917 24.777,10.415 21.949,7.585 16.447,13.087 10.945,7.585 8.117,10.415 13.618,15.917 8.116,21.419 10.946,24.248 16.447,18.746 21.948,24.248z"><title>Remove vertex.</title></path></g></g><g  id="v-661" display="none"><g  id="v-641"><path  end="source" d="M 26 0 L 0 13 L 26 26 z"></path></g><g  id="v-642"><path  end="target" d="M 26 0 L 0 13 L 26 26 z"></path></g></g><g  id="v-660" display="none"><g  id="v-639" transform="translate(1027.6649169921875, 1096.943603515625) "><g  event="remove"><circle r="11"></circle><path transform="scale(.8) translate(-16, -16)" d="M24.778,21.419 19.276,15.917 24.777,10.415 21.949,7.585 16.447,13.087 10.945,7.585 8.117,10.415 13.618,15.917 8.116,21.419 10.946,24.248 16.447,18.746 21.948,24.248z"></path><title>Remove link.</title></g><g  event="link:options"><circle r="11" transform="translate(25)"></circle><path fill="white" transform="scale(.55) translate(29, -16)" d="M31.229,17.736c0.064-0.571,0.104-1.148,0.104-1.736s-0.04-1.166-0.104-1.737l-4.377-1.557c-0.218-0.716-0.504-1.401-0.851-2.05l1.993-4.192c-0.725-0.91-1.549-1.734-2.458-2.459l-4.193,1.994c-0.647-0.347-1.334-0.632-2.049-0.849l-1.558-4.378C17.165,0.708,16.588,0.667,16,0.667s-1.166,0.041-1.737,0.105L12.707,5.15c-0.716,0.217-1.401,0.502-2.05,0.849L6.464,4.005C5.554,4.73,4.73,5.554,4.005,6.464l1.994,4.192c-0.347,0.648-0.632,1.334-0.849,2.05l-4.378,1.557C0.708,14.834,0.667,15.412,0.667,16s0.041,1.165,0.105,1.736l4.378,1.558c0.217,0.715,0.502,1.401,0.849,2.049l-1.994,4.193c0.725,0.909,1.549,1.733,2.459,2.458l4.192-1.993c0.648,0.347,1.334,0.633,2.05,0.851l1.557,4.377c0.571,0.064,1.148,0.104,1.737,0.104c0.588,0,1.165-0.04,1.736-0.104l1.558-4.377c0.715-0.218,1.399-0.504,2.049-0.851l4.193,1.993c0.909-0.725,1.733-1.549,2.458-2.458l-1.993-4.193c0.347-0.647,0.633-1.334,0.851-2.049L31.229,17.736zM16,20.871c-2.69,0-4.872-2.182-4.872-4.871c0-2.69,2.182-4.872,4.872-4.872c2.689,0,4.871,2.182,4.871,4.872C20.871,18.689,18.689,20.871,16,20.871z"></path><title>Link options.</title></g></g></g></g></g><defs id="v-4"></defs></svg>
\ No newline at end of file

From 8f2fc6be9cc241a1e3e1d3f46bd4a4bbc9ec6d86 Mon Sep 17 00:00:00 2001
From: Ishant Thakare <ishantrog752@gmail.com>
Date: Fri, 17 Jan 2025 08:29:21 +0530
Subject: [PATCH 5/6] Fix code format

Signed-off-by: Ishant Thakare <ishantrog752@gmail.com>
---
 openfl/component/aggregator/straggler_handling.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/openfl/component/aggregator/straggler_handling.py b/openfl/component/aggregator/straggler_handling.py
index 1d5f6ef7de..d1157ae3ff 100644
--- a/openfl/component/aggregator/straggler_handling.py
+++ b/openfl/component/aggregator/straggler_handling.py
@@ -83,7 +83,7 @@ def __init__(
         self.is_timer_started = False
 
         if self.straggler_cutoff_time == np.inf:
-            logger.warning("CutoffPolicy is disabled as straggler_cutoff_time " "is set to np.inf.")
+            logger.warning("CutoffPolicy is disabled as straggler_cutoff_time is set to np.inf.")
 
     def reset_policy_for_round(self) -> None:
         """Reset timer for the next round."""

From 3133cb68965f8edd07d5b1c8e3260f187fb2fe58 Mon Sep 17 00:00:00 2001
From: Ishant Thakare <ishantrog752@gmail.com>
Date: Wed, 22 Jan 2025 08:36:11 +0530
Subject: [PATCH 6/6] Incorporated review comments

Signed-off-by: Ishant Thakare <ishantrog752@gmail.com>
---
 .../straggler_handling_algorithms.rst               |  6 +++---
 openfl/component/__init__.py                        |  2 +-
 openfl/component/aggregator/__init__.py             |  2 +-
 openfl/component/aggregator/aggregator.py           |  4 ++--
 openfl/component/aggregator/straggler_handling.py   | 13 ++++++++-----
 openfl/federated/plan/plan.py                       |  2 +-
 6 files changed, 16 insertions(+), 13 deletions(-)

diff --git a/docs/developer_guide/advanced_topics/straggler_handling_algorithms.rst b/docs/developer_guide/advanced_topics/straggler_handling_algorithms.rst
index 32489e9ef1..88579bde57 100644
--- a/docs/developer_guide/advanced_topics/straggler_handling_algorithms.rst
+++ b/docs/developer_guide/advanced_topics/straggler_handling_algorithms.rst
@@ -11,7 +11,7 @@ The Open Federated Learning (OpenFL) framework supports straggler handling inter
 
 The following are the straggler handling algorithms supported in OpenFL:
 
-``CutoffPolicy``
+``CutoffTimePolicy``
     Identifies stragglers based on the cutoff time specified in the settings. Arguments to the function are:
         - *Cutoff Time* (straggler_cutoff_time), specifies the cutoff time by which the aggregator should end the round early.
         - *Minimum Reporting* (minimum_reporting), specifies the minimum number of collaborators needed to aggregate the model.
@@ -29,12 +29,12 @@ The following are the straggler handling algorithms supported in OpenFL:
 Demonstration of adding the straggler handling interface
 =========================================================
 
-The example template, **torch_cnn_mnist_straggler_check**, uses the ``PercentagePolicy``. To gain a better understanding of how experiments perform, you can modify the **percent_collaborators_needed** or **minimum_reporting** parameter in the template **plan.yaml** or even choose **CutoffPolicy** function instead:
+The example template, **torch_cnn_mnist_straggler_check**, uses the ``PercentagePolicy``. To gain a better understanding of how experiments perform, you can modify the **percent_collaborators_needed** or **minimum_reporting** parameter in the template **plan.yaml** or even choose **CutoffTimePolicy** function instead:
 
     .. code-block:: yaml
     
         straggler_handling_policy :
-            template : openfl.component.aggregator.straggler_handling.CutoffPolicy
+            template : openfl.component.aggregator.straggler_handling.CutoffTimePolicy
             settings :
                 straggler_cutoff_time : 20
                 minimum_reporting : 1
diff --git a/openfl/component/__init__.py b/openfl/component/__init__.py
index 792cfd855d..499af56320 100644
--- a/openfl/component/__init__.py
+++ b/openfl/component/__init__.py
@@ -4,7 +4,7 @@
 
 from openfl.component.aggregator.aggregator import Aggregator
 from openfl.component.aggregator.straggler_handling import (
-    CutoffPolicy,
+    CutoffTimePolicy,
     PercentagePolicy,
     StragglerPolicy,
 )
diff --git a/openfl/component/aggregator/__init__.py b/openfl/component/aggregator/__init__.py
index 1bbddaf4f2..71a7ce11ea 100644
--- a/openfl/component/aggregator/__init__.py
+++ b/openfl/component/aggregator/__init__.py
@@ -4,7 +4,7 @@
 
 from openfl.component.aggregator.aggregator import Aggregator
 from openfl.component.aggregator.straggler_handling import (
-    CutoffPolicy,
+    CutoffTimePolicy,
     PercentagePolicy,
     StragglerPolicy,
 )
diff --git a/openfl/component/aggregator/aggregator.py b/openfl/component/aggregator/aggregator.py
index e48039875e..5ede1c1da9 100644
--- a/openfl/component/aggregator/aggregator.py
+++ b/openfl/component/aggregator/aggregator.py
@@ -10,7 +10,7 @@
 from typing import List, Optional
 
 import openfl.callbacks as callbacks_module
-from openfl.component.aggregator.straggler_handling import CutoffPolicy, StragglerPolicy
+from openfl.component.aggregator.straggler_handling import CutoffTimePolicy, StragglerPolicy
 from openfl.databases import PersistentTensorDB, TensorDB
 from openfl.interface.aggregation_functions import WeightedAverage
 from openfl.pipelines import NoCompressionPipeline, TensorCodec
@@ -73,7 +73,7 @@ def __init__(
         last_state_path,
         assigner,
         use_delta_updates=True,
-        straggler_handling_policy: StragglerPolicy = CutoffPolicy,
+        straggler_handling_policy: StragglerPolicy = CutoffTimePolicy,
         rounds_to_train=256,
         single_col_cert_common_name=None,
         compression_pipeline=None,
diff --git a/openfl/component/aggregator/straggler_handling.py b/openfl/component/aggregator/straggler_handling.py
index d1157ae3ff..66207bddf1 100644
--- a/openfl/component/aggregator/straggler_handling.py
+++ b/openfl/component/aggregator/straggler_handling.py
@@ -22,7 +22,7 @@ class StragglerPolicy(ABC):
     def start_policy(self, **kwargs) -> None:
         """
         Start straggler handling policy for collaborator for a particular round.
-        NOTE: Refer CutoffPolicy for reference.
+        NOTE: Refer CutoffTimePolicy for reference.
 
         Args:
             **kwargs
@@ -56,14 +56,14 @@ def straggler_cutoff_check(
         raise NotImplementedError
 
 
-class CutoffPolicy(StragglerPolicy):
+class CutoffTimePolicy(StragglerPolicy):
     """Cutoff time based Straggler Handling function."""
 
     def __init__(
         self, round_start_time=None, straggler_cutoff_time=np.inf, minimum_reporting=1, **kwargs
     ):
         """
-         Initialize a CutoffPolicy object.
+         Initialize a CutoffTimePolicy object.
 
         Args:
             round_start_time (optional): The start time of the round. Defaults
@@ -71,7 +71,8 @@ def __init__(
             straggler_cutoff_time (float, optional): The cutoff time for
                 stragglers. Defaults to np.inf.
             minimum_reporting (int, optional): The minimum number of
-                collaborators that should report. Defaults to 1.
+                collaborators that should report before moving to the next round.
+                Defaults to 1.
             **kwargs: Variable length argument list.
         """
         if minimum_reporting <= 0:
@@ -83,7 +84,9 @@ def __init__(
         self.is_timer_started = False
 
         if self.straggler_cutoff_time == np.inf:
-            logger.warning("CutoffPolicy is disabled as straggler_cutoff_time is set to np.inf.")
+            logger.warning(
+                "CutoffTimePolicy is disabled as straggler_cutoff_time is set to np.inf."
+            )
 
     def reset_policy_for_round(self) -> None:
         """Reset timer for the next round."""
diff --git a/openfl/federated/plan/plan.py b/openfl/federated/plan/plan.py
index 0fe1e87c41..e26bfa56d8 100644
--- a/openfl/federated/plan/plan.py
+++ b/openfl/federated/plan/plan.py
@@ -411,7 +411,7 @@ def get_tensor_pipe(self):
 
     def get_straggler_handling_policy(self):
         """Get straggler handling policy."""
-        template = "openfl.component.aggregator.straggler_handling.CutoffPolicy"
+        template = "openfl.component.aggregator.straggler_handling.CutoffTimePolicy"
         defaults = self.config.get("straggler_handling_policy", {TEMPLATE: template, SETTINGS: {}})
 
         if self.straggler_policy_ is None: