flwr-nightly 1.7.0.dev20240103__py3-none-any.whl → 1.7.0.dev20240104__py3-none-any.whl

flwr/client/app.py

@@ -349,7 +349,7 @@ def _start_client_internal(
                     break
 
                 # Register state
-                node_state.register_workloadstate(workload_id=task_ins.workload_id)
+                node_state.register_workloadstate(run_id=task_ins.run_id)
 
                 # Load app
                 app: Flower = load_flower_callable_fn()
@@ -357,15 +357,13 @@ def _start_client_internal(
                 # Handle task message
                 fwd_msg: Fwd = Fwd(
                     task_ins=task_ins,
-                    state=node_state.retrieve_workloadstate(
-                        workload_id=task_ins.workload_id
-                    ),
+                    state=node_state.retrieve_workloadstate(run_id=task_ins.run_id),
                 )
                 bwd_msg: Bwd = app(fwd=fwd_msg)
 
                 # Update node state
                 node_state.update_workloadstate(
-                    workload_id=bwd_msg.task_res.workload_id,
+                    run_id=bwd_msg.task_res.run_id,
                     workload_state=bwd_msg.state,
                 )
 

flwr/client/grpc_client/connection.py

@@ -119,7 +119,7 @@ def grpc_connection(
         return TaskIns(
             task_id=str(uuid.uuid4()),
             group_id="",
-            workload_id=0,
+            run_id=0,
             task=Task(
                 producer=Node(node_id=0, anonymous=True),
                 consumer=Node(node_id=0, anonymous=True),

flwr/client/message_handler/message_handler.py

@@ -112,7 +112,7 @@ def handle(
             task_res = TaskRes(
                 task_id="",
                 group_id="",
-                workload_id=0,
+                run_id=0,
                 task=Task(
                     ancestry=[],
                     sa=SecureAggregation(named_values=serde.named_values_to_proto(res)),

flwr/client/message_handler/task_handler.py

@@ -70,7 +70,7 @@ def validate_task_res(task_res: TaskRes) -> bool:
     Returns
     -------
     is_valid: bool
-        True if the `task_id`, `group_id`, and `workload_id` fields in TaskRes
+        True if the `task_id`, `group_id`, and `run_id` fields in TaskRes
         and the `producer`, `consumer`, and `ancestry` fields in its sub-message Task
         are not initialized accidentally elsewhere,
         False otherwise.
@@ -84,7 +84,7 @@ def validate_task_res(task_res: TaskRes) -> bool:
     if (
         "task_id" in initialized_fields_in_task_res
         or "group_id" in initialized_fields_in_task_res
-        or "workload_id" in initialized_fields_in_task_res
+        or "run_id" in initialized_fields_in_task_res
         or "producer" in initialized_fields_in_task
         or "consumer" in initialized_fields_in_task
         or "ancestry" in initialized_fields_in_task
@@ -129,7 +129,7 @@ def wrap_client_message_in_task_res(client_message: ClientMessage) -> TaskRes:
     return TaskRes(
         task_id="",
         group_id="",
-        workload_id=0,
+        run_id=0,
         task=Task(ancestry=[], legacy_client_message=client_message),
     )
 
@@ -139,7 +139,7 @@ def configure_task_res(
 ) -> TaskRes:
     """Set the metadata of a TaskRes.
 
-    Fill `group_id` and `workload_id` in TaskRes
+    Fill `group_id` and `run_id` in TaskRes
     and `producer`, `consumer`, and `ancestry` in Task in TaskRes.
 
     `producer` in Task in TaskRes will remain unchanged/unset.
@@ -152,7 +152,7 @@ def configure_task_res(
     task_res = TaskRes(
         task_id="",  # This will be generated by the server
         group_id=ref_task_ins.group_id,
-        workload_id=ref_task_ins.workload_id,
+        run_id=ref_task_ins.run_id,
         task=task_res.task,
     )
     # pylint: disable-next=no-member

flwr/client/node_state.py

@@ -27,24 +27,22 @@ class NodeState:
         self._meta: Dict[str, Any] = {}  # holds metadata about the node
         self.workload_states: Dict[int, WorkloadState] = {}
 
-    def register_workloadstate(self, workload_id: int) -> None:
+    def register_workloadstate(self, run_id: int) -> None:
         """Register new workload state for this node."""
-        if workload_id not in self.workload_states:
-            self.workload_states[workload_id] = WorkloadState({})
+        if run_id not in self.workload_states:
+            self.workload_states[run_id] = WorkloadState({})
 
-    def retrieve_workloadstate(self, workload_id: int) -> WorkloadState:
-        """Get workload state given a workload_id."""
-        if workload_id in self.workload_states:
-            return self.workload_states[workload_id]
+    def retrieve_workloadstate(self, run_id: int) -> WorkloadState:
+        """Get workload state given a run_id."""
+        if run_id in self.workload_states:
+            return self.workload_states[run_id]
 
         raise RuntimeError(
-            f"WorkloadState for workload_id={workload_id} doesn't exist."
+            f"WorkloadState for run_id={run_id} doesn't exist."
             " A workload must be registered before it can be retrieved or updated "
             " by a client."
         )
 
-    def update_workloadstate(
-        self, workload_id: int, workload_state: WorkloadState
-    ) -> None:
+    def update_workloadstate(self, run_id: int, workload_state: WorkloadState) -> None:
         """Update workload state."""
-        self.workload_states[workload_id] = workload_state
+        self.workload_states[run_id] = workload_state

flwr/client/node_state_tests.py

@@ -32,7 +32,7 @@ def _run_dummy_task(state: WorkloadState) -> WorkloadState:
 def test_multiworkload_in_node_state() -> None:
     """Test basic NodeState logic."""
     # Tasks to perform
-    tasks = [TaskIns(workload_id=w_id) for w_id in [0, 1, 1, 2, 3, 2, 1, 5]]
+    tasks = [TaskIns(run_id=r_id) for r_id in [0, 1, 1, 2, 3, 2, 1, 5]]
     # the "tasks" is to count how many times each workload is executed
     expected_values = {0: "1", 1: "1" * 3, 2: "1" * 2, 3: "1", 5: "1"}
 
@@ -40,20 +40,20 @@ def test_multiworkload_in_node_state() -> None:
     node_state = NodeState()
 
     for task in tasks:
-        w_id = task.workload_id
+        r_id = task.run_id
 
         # Register
-        node_state.register_workloadstate(workload_id=w_id)
+        node_state.register_workloadstate(run_id=r_id)
 
         # Get workload state
-        state = node_state.retrieve_workloadstate(workload_id=w_id)
+        state = node_state.retrieve_workloadstate(run_id=r_id)
 
         # Run "task"
         updated_state = _run_dummy_task(state)
 
         # Update workload state
-        node_state.update_workloadstate(workload_id=w_id, workload_state=updated_state)
+        node_state.update_workloadstate(run_id=r_id, workload_state=updated_state)
 
     # Verify values
-    for w_id, state in node_state.workload_states.items():
-        assert state.state["counter"] == expected_values[w_id]
+    for r_id, state in node_state.workload_states.items():
+        assert state.state["counter"] == expected_values[r_id]

flwr/driver/app.py

@@ -170,8 +170,8 @@ def update_client_manager(
     and dead nodes will be removed from the ClientManager via
     `client_manager.unregister()`.
     """
-    # Request for workload_id
-    workload_id = driver.create_workload(driver_pb2.CreateWorkloadRequest()).workload_id
+    # Request for run_id
+    run_id = driver.create_workload(driver_pb2.CreateWorkloadRequest()).run_id
 
     # Loop until the driver is disconnected
     registered_nodes: Dict[int, DriverClientProxy] = {}
@@ -181,7 +181,7 @@ def update_client_manager(
             if driver.stub is None:
                 break
             get_nodes_res = driver.get_nodes(
-                req=driver_pb2.GetNodesRequest(workload_id=workload_id)
+                req=driver_pb2.GetNodesRequest(run_id=run_id)
             )
         all_node_ids = {node.node_id for node in get_nodes_res.nodes}
         dead_nodes = set(registered_nodes).difference(all_node_ids)
@@ -199,7 +199,7 @@ def update_client_manager(
                 node_id=node_id,
                 driver=driver,
                 anonymous=False,
-                workload_id=workload_id,
+                run_id=run_id,
             )
             if client_manager.register(client_proxy):
                 registered_nodes[node_id] = client_proxy

flwr/driver/driver.py

@@ -54,37 +54,37 @@ class Driver:
         self.addr = driver_service_address
         self.certificates = certificates
         self.grpc_driver: Optional[GrpcDriver] = None
-        self.workload_id: Optional[int] = None
+        self.run_id: Optional[int] = None
         self.node = Node(node_id=0, anonymous=True)
 
-    def _get_grpc_driver_and_workload_id(self) -> Tuple[GrpcDriver, int]:
+    def _get_grpc_driver_and_run_id(self) -> Tuple[GrpcDriver, int]:
         # Check if the GrpcDriver is initialized
-        if self.grpc_driver is None or self.workload_id is None:
+        if self.grpc_driver is None or self.run_id is None:
             # Connect and create workload
             self.grpc_driver = GrpcDriver(
                 driver_service_address=self.addr, certificates=self.certificates
             )
             self.grpc_driver.connect()
             res = self.grpc_driver.create_workload(CreateWorkloadRequest())
-            self.workload_id = res.workload_id
+            self.run_id = res.run_id
 
-        return self.grpc_driver, self.workload_id
+        return self.grpc_driver, self.run_id
 
     def get_nodes(self) -> List[Node]:
         """Get node IDs."""
-        grpc_driver, workload_id = self._get_grpc_driver_and_workload_id()
+        grpc_driver, run_id = self._get_grpc_driver_and_run_id()
 
         # Call GrpcDriver method
-        res = grpc_driver.get_nodes(GetNodesRequest(workload_id=workload_id))
+        res = grpc_driver.get_nodes(GetNodesRequest(run_id=run_id))
         return list(res.nodes)
 
     def push_task_ins(self, task_ins_list: List[TaskIns]) -> List[str]:
         """Schedule tasks."""
-        grpc_driver, workload_id = self._get_grpc_driver_and_workload_id()
+        grpc_driver, run_id = self._get_grpc_driver_and_run_id()
 
-        # Set workload_id
+        # Set run_id
         for task_ins in task_ins_list:
-            task_ins.workload_id = workload_id
+            task_ins.run_id = run_id
 
         # Call GrpcDriver method
         res = grpc_driver.push_task_ins(PushTaskInsRequest(task_ins_list=task_ins_list))
@@ -92,7 +92,7 @@ class Driver:
 
     def pull_task_res(self, task_ids: Iterable[str]) -> List[TaskRes]:
         """Get task results."""
-        grpc_driver, _ = self._get_grpc_driver_and_workload_id()
+        grpc_driver, _ = self._get_grpc_driver_and_run_id()
 
         # Call GrpcDriver method
         res = grpc_driver.pull_task_res(

flwr/driver/driver_client_proxy.py

@@ -31,13 +31,11 @@ SLEEP_TIME = 1
 class DriverClientProxy(ClientProxy):
     """Flower client proxy which delegates work using the Driver API."""
 
-    def __init__(
-        self, node_id: int, driver: GrpcDriver, anonymous: bool, workload_id: int
-    ):
+    def __init__(self, node_id: int, driver: GrpcDriver, anonymous: bool, run_id: int):
         super().__init__(str(node_id))
         self.node_id = node_id
         self.driver = driver
-        self.workload_id = workload_id
+        self.run_id = run_id
         self.anonymous = anonymous
 
     def get_properties(
@@ -106,7 +104,7 @@ class DriverClientProxy(ClientProxy):
         task_ins = task_pb2.TaskIns(
             task_id="",
             group_id="",
-            workload_id=self.workload_id,
+            run_id=self.run_id,
             task=task_pb2.Task(
                 producer=node_pb2.Node(
                     node_id=0,

flwr/proto/driver_pb2.py

@@ -16,7 +16,7 @@ from flwr.proto import node_pb2 as flwr_dot_proto_dot_node__pb2
 from flwr.proto import task_pb2 as flwr_dot_proto_dot_task__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x17\x66lwr/proto/driver.proto\x12\nflwr.proto\x1a\x15\x66lwr/proto/node.proto\x1a\x15\x66lwr/proto/task.proto\"\x17\n\x15\x43reateWorkloadRequest\"-\n\x16\x43reateWorkloadResponse\x12\x13\n\x0bworkload_id\x18\x01 \x01(\x12\"&\n\x0fGetNodesRequest\x12\x13\n\x0bworkload_id\x18\x01 \x01(\x12\"3\n\x10GetNodesResponse\x12\x1f\n\x05nodes\x18\x01 \x03(\x0b\x32\x10.flwr.proto.Node\"@\n\x12PushTaskInsRequest\x12*\n\rtask_ins_list\x18\x01 \x03(\x0b\x32\x13.flwr.proto.TaskIns\"\'\n\x13PushTaskInsResponse\x12\x10\n\x08task_ids\x18\x02 \x03(\t\"F\n\x12PullTaskResRequest\x12\x1e\n\x04node\x18\x01 \x01(\x0b\x32\x10.flwr.proto.Node\x12\x10\n\x08task_ids\x18\x02 \x03(\t\"A\n\x13PullTaskResResponse\x12*\n\rtask_res_list\x18\x01 \x03(\x0b\x32\x13.flwr.proto.TaskRes2\xd0\x02\n\x06\x44river\x12Y\n\x0e\x43reateWorkload\x12!.flwr.proto.CreateWorkloadRequest\x1a\".flwr.proto.CreateWorkloadResponse\"\x00\x12G\n\x08GetNodes\x12\x1b.flwr.proto.GetNodesRequest\x1a\x1c.flwr.proto.GetNodesResponse\"\x00\x12P\n\x0bPushTaskIns\x12\x1e.flwr.proto.PushTaskInsRequest\x1a\x1f.flwr.proto.PushTaskInsResponse\"\x00\x12P\n\x0bPullTaskRes\x12\x1e.flwr.proto.PullTaskResRequest\x1a\x1f.flwr.proto.PullTaskResResponse\"\x00\x62\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x17\x66lwr/proto/driver.proto\x12\nflwr.proto\x1a\x15\x66lwr/proto/node.proto\x1a\x15\x66lwr/proto/task.proto\"\x17\n\x15\x43reateWorkloadRequest\"(\n\x16\x43reateWorkloadResponse\x12\x0e\n\x06run_id\x18\x01 \x01(\x12\"!\n\x0fGetNodesRequest\x12\x0e\n\x06run_id\x18\x01 \x01(\x12\"3\n\x10GetNodesResponse\x12\x1f\n\x05nodes\x18\x01 \x03(\x0b\x32\x10.flwr.proto.Node\"@\n\x12PushTaskInsRequest\x12*\n\rtask_ins_list\x18\x01 \x03(\x0b\x32\x13.flwr.proto.TaskIns\"\'\n\x13PushTaskInsResponse\x12\x10\n\x08task_ids\x18\x02 \x03(\t\"F\n\x12PullTaskResRequest\x12\x1e\n\x04node\x18\x01 \x01(\x0b\x32\x10.flwr.proto.Node\x12\x10\n\x08task_ids\x18\x02 \x03(\t\"A\n\x13PullTaskResResponse\x12*\n\rtask_res_list\x18\x01 \x03(\x0b\x32\x13.flwr.proto.TaskRes2\xd0\x02\n\x06\x44river\x12Y\n\x0e\x43reateWorkload\x12!.flwr.proto.CreateWorkloadRequest\x1a\".flwr.proto.CreateWorkloadResponse\"\x00\x12G\n\x08GetNodes\x12\x1b.flwr.proto.GetNodesRequest\x1a\x1c.flwr.proto.GetNodesResponse\"\x00\x12P\n\x0bPushTaskIns\x12\x1e.flwr.proto.PushTaskInsRequest\x1a\x1f.flwr.proto.PushTaskInsResponse\"\x00\x12P\n\x0bPullTaskRes\x12\x1e.flwr.proto.PullTaskResRequest\x1a\x1f.flwr.proto.PullTaskResResponse\"\x00\x62\x06proto3')
 
 
 
@@ -91,19 +91,19 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _CREATEWORKLOADREQUEST._serialized_start=85
   _CREATEWORKLOADREQUEST._serialized_end=108
   _CREATEWORKLOADRESPONSE._serialized_start=110
-  _CREATEWORKLOADRESPONSE._serialized_end=155
-  _GETNODESREQUEST._serialized_start=157
-  _GETNODESREQUEST._serialized_end=195
-  _GETNODESRESPONSE._serialized_start=197
-  _GETNODESRESPONSE._serialized_end=248
-  _PUSHTASKINSREQUEST._serialized_start=250
-  _PUSHTASKINSREQUEST._serialized_end=314
-  _PUSHTASKINSRESPONSE._serialized_start=316
-  _PUSHTASKINSRESPONSE._serialized_end=355
-  _PULLTASKRESREQUEST._serialized_start=357
-  _PULLTASKRESREQUEST._serialized_end=427
-  _PULLTASKRESRESPONSE._serialized_start=429
-  _PULLTASKRESRESPONSE._serialized_end=494
-  _DRIVER._serialized_start=497
-  _DRIVER._serialized_end=833
+  _CREATEWORKLOADRESPONSE._serialized_end=150
+  _GETNODESREQUEST._serialized_start=152
+  _GETNODESREQUEST._serialized_end=185
+  _GETNODESRESPONSE._serialized_start=187
+  _GETNODESRESPONSE._serialized_end=238
+  _PUSHTASKINSREQUEST._serialized_start=240
+  _PUSHTASKINSREQUEST._serialized_end=304
+  _PUSHTASKINSRESPONSE._serialized_start=306
+  _PUSHTASKINSRESPONSE._serialized_end=345
+  _PULLTASKRESREQUEST._serialized_start=347
+  _PULLTASKRESREQUEST._serialized_end=417
+  _PULLTASKRESRESPONSE._serialized_start=419
+  _PULLTASKRESRESPONSE._serialized_end=484
+  _DRIVER._serialized_start=487
+  _DRIVER._serialized_end=823
 # @@protoc_insertion_point(module_scope)

flwr/proto/driver_pb2.pyi

@@ -22,25 +22,25 @@ global___CreateWorkloadRequest = CreateWorkloadRequest
 
 class CreateWorkloadResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
-    WORKLOAD_ID_FIELD_NUMBER: builtins.int
-    workload_id: builtins.int
+    RUN_ID_FIELD_NUMBER: builtins.int
+    run_id: builtins.int
     def __init__(self,
         *,
-        workload_id: builtins.int = ...,
+        run_id: builtins.int = ...,
         ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["workload_id",b"workload_id"]) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["run_id",b"run_id"]) -> None: ...
 global___CreateWorkloadResponse = CreateWorkloadResponse
 
 class GetNodesRequest(google.protobuf.message.Message):
     """GetNodes messages"""
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
-    WORKLOAD_ID_FIELD_NUMBER: builtins.int
-    workload_id: builtins.int
+    RUN_ID_FIELD_NUMBER: builtins.int
+    run_id: builtins.int
     def __init__(self,
         *,
-        workload_id: builtins.int = ...,
+        run_id: builtins.int = ...,
         ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["workload_id",b"workload_id"]) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["run_id",b"run_id"]) -> None: ...
 global___GetNodesRequest = GetNodesRequest
 
 class GetNodesResponse(google.protobuf.message.Message):

flwr/proto/task_pb2.py

@@ -16,7 +16,7 @@ from flwr.proto import node_pb2 as flwr_dot_proto_dot_node__pb2
 from flwr.proto import transport_pb2 as flwr_dot_proto_dot_transport__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x15\x66lwr/proto/task.proto\x12\nflwr.proto\x1a\x15\x66lwr/proto/node.proto\x1a\x1a\x66lwr/proto/transport.proto\"\xbe\x02\n\x04Task\x12\"\n\x08producer\x18\x01 \x01(\x0b\x32\x10.flwr.proto.Node\x12\"\n\x08\x63onsumer\x18\x02 \x01(\x0b\x32\x10.flwr.proto.Node\x12\x12\n\ncreated_at\x18\x03 \x01(\t\x12\x14\n\x0c\x64\x65livered_at\x18\x04 \x01(\t\x12\x0b\n\x03ttl\x18\x05 \x01(\t\x12\x10\n\x08\x61ncestry\x18\x06 \x03(\t\x12)\n\x02sa\x18\x07 \x01(\x0b\x32\x1d.flwr.proto.SecureAggregation\x12<\n\x15legacy_server_message\x18\x65 \x01(\x0b\x32\x19.flwr.proto.ServerMessageB\x02\x18\x01\x12<\n\x15legacy_client_message\x18\x66 \x01(\x0b\x32\x19.flwr.proto.ClientMessageB\x02\x18\x01\"a\n\x07TaskIns\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x10\n\x08group_id\x18\x02 \x01(\t\x12\x13\n\x0bworkload_id\x18\x03 \x01(\x12\x12\x1e\n\x04task\x18\x04 \x01(\x0b\x32\x10.flwr.proto.Task\"a\n\x07TaskRes\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x10\n\x08group_id\x18\x02 \x01(\t\x12\x13\n\x0bworkload_id\x18\x03 \x01(\x12\x12\x1e\n\x04task\x18\x04 \x01(\x0b\x32\x10.flwr.proto.Task\"\xf3\x03\n\x05Value\x12\x10\n\x06\x64ouble\x18\x01 \x01(\x01H\x00\x12\x10\n\x06sint64\x18\x02 \x01(\x12H\x00\x12\x0e\n\x04\x62ool\x18\x03 \x01(\x08H\x00\x12\x10\n\x06string\x18\x04 \x01(\tH\x00\x12\x0f\n\x05\x62ytes\x18\x05 \x01(\x0cH\x00\x12\x33\n\x0b\x64ouble_list\x18\x15 \x01(\x0b\x32\x1c.flwr.proto.Value.DoubleListH\x00\x12\x33\n\x0bsint64_list\x18\x16 \x01(\x0b\x32\x1c.flwr.proto.Value.Sint64ListH\x00\x12/\n\tbool_list\x18\x17 \x01(\x0b\x32\x1a.flwr.proto.Value.BoolListH\x00\x12\x33\n\x0bstring_list\x18\x18 \x01(\x0b\x32\x1c.flwr.proto.Value.StringListH\x00\x12\x31\n\nbytes_list\x18\x19 \x01(\x0b\x32\x1b.flwr.proto.Value.BytesListH\x00\x1a\x1a\n\nDoubleList\x12\x0c\n\x04vals\x18\x01 \x03(\x01\x1a\x1a\n\nSint64List\x12\x0c\n\x04vals\x18\x01 \x03(\x12\x1a\x18\n\x08\x42oolList\x12\x0c\n\x04vals\x18\x01 \x03(\x08\x1a\x1a\n\nStringList\x12\x0c\n\x04vals\x18\x01 \x03(\t\x1a\x19\n\tBytesList\x12\x0c\n\x04vals\x18\x01 \x03(\x0c\x42\x07\n\x05value\"\xa0\x01\n\x11SecureAggregation\x12\x44\n\x0cnamed_values\x18\x01 \x03(\x0b\x32..flwr.proto.SecureAggregation.NamedValuesEntry\x1a\x45\n\x10NamedValuesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12 \n\x05value\x18\x02 \x01(\x0b\x32\x11.flwr.proto.Value:\x02\x38\x01\x62\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x15\x66lwr/proto/task.proto\x12\nflwr.proto\x1a\x15\x66lwr/proto/node.proto\x1a\x1a\x66lwr/proto/transport.proto\"\xbe\x02\n\x04Task\x12\"\n\x08producer\x18\x01 \x01(\x0b\x32\x10.flwr.proto.Node\x12\"\n\x08\x63onsumer\x18\x02 \x01(\x0b\x32\x10.flwr.proto.Node\x12\x12\n\ncreated_at\x18\x03 \x01(\t\x12\x14\n\x0c\x64\x65livered_at\x18\x04 \x01(\t\x12\x0b\n\x03ttl\x18\x05 \x01(\t\x12\x10\n\x08\x61ncestry\x18\x06 \x03(\t\x12)\n\x02sa\x18\x07 \x01(\x0b\x32\x1d.flwr.proto.SecureAggregation\x12<\n\x15legacy_server_message\x18\x65 \x01(\x0b\x32\x19.flwr.proto.ServerMessageB\x02\x18\x01\x12<\n\x15legacy_client_message\x18\x66 \x01(\x0b\x32\x19.flwr.proto.ClientMessageB\x02\x18\x01\"\\\n\x07TaskIns\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x10\n\x08group_id\x18\x02 \x01(\t\x12\x0e\n\x06run_id\x18\x03 \x01(\x12\x12\x1e\n\x04task\x18\x04 \x01(\x0b\x32\x10.flwr.proto.Task\"\\\n\x07TaskRes\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x10\n\x08group_id\x18\x02 \x01(\t\x12\x0e\n\x06run_id\x18\x03 \x01(\x12\x12\x1e\n\x04task\x18\x04 \x01(\x0b\x32\x10.flwr.proto.Task\"\xf3\x03\n\x05Value\x12\x10\n\x06\x64ouble\x18\x01 \x01(\x01H\x00\x12\x10\n\x06sint64\x18\x02 \x01(\x12H\x00\x12\x0e\n\x04\x62ool\x18\x03 \x01(\x08H\x00\x12\x10\n\x06string\x18\x04 \x01(\tH\x00\x12\x0f\n\x05\x62ytes\x18\x05 \x01(\x0cH\x00\x12\x33\n\x0b\x64ouble_list\x18\x15 \x01(\x0b\x32\x1c.flwr.proto.Value.DoubleListH\x00\x12\x33\n\x0bsint64_list\x18\x16 \x01(\x0b\x32\x1c.flwr.proto.Value.Sint64ListH\x00\x12/\n\tbool_list\x18\x17 \x01(\x0b\x32\x1a.flwr.proto.Value.BoolListH\x00\x12\x33\n\x0bstring_list\x18\x18 \x01(\x0b\x32\x1c.flwr.proto.Value.StringListH\x00\x12\x31\n\nbytes_list\x18\x19 \x01(\x0b\x32\x1b.flwr.proto.Value.BytesListH\x00\x1a\x1a\n\nDoubleList\x12\x0c\n\x04vals\x18\x01 \x03(\x01\x1a\x1a\n\nSint64List\x12\x0c\n\x04vals\x18\x01 \x03(\x12\x1a\x18\n\x08\x42oolList\x12\x0c\n\x04vals\x18\x01 \x03(\x08\x1a\x1a\n\nStringList\x12\x0c\n\x04vals\x18\x01 \x03(\t\x1a\x19\n\tBytesList\x12\x0c\n\x04vals\x18\x01 \x03(\x0c\x42\x07\n\x05value\"\xa0\x01\n\x11SecureAggregation\x12\x44\n\x0cnamed_values\x18\x01 \x03(\x0b\x32..flwr.proto.SecureAggregation.NamedValuesEntry\x1a\x45\n\x10NamedValuesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12 \n\x05value\x18\x02 \x01(\x0b\x32\x11.flwr.proto.Value:\x02\x38\x01\x62\x06proto3')
 
 
 
@@ -126,23 +126,23 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _TASK._serialized_start=89
   _TASK._serialized_end=407
   _TASKINS._serialized_start=409
-  _TASKINS._serialized_end=506
-  _TASKRES._serialized_start=508
-  _TASKRES._serialized_end=605
-  _VALUE._serialized_start=608
-  _VALUE._serialized_end=1107
-  _VALUE_DOUBLELIST._serialized_start=963
-  _VALUE_DOUBLELIST._serialized_end=989
-  _VALUE_SINT64LIST._serialized_start=991
-  _VALUE_SINT64LIST._serialized_end=1017
-  _VALUE_BOOLLIST._serialized_start=1019
-  _VALUE_BOOLLIST._serialized_end=1043
-  _VALUE_STRINGLIST._serialized_start=1045
-  _VALUE_STRINGLIST._serialized_end=1071
-  _VALUE_BYTESLIST._serialized_start=1073
-  _VALUE_BYTESLIST._serialized_end=1098
-  _SECUREAGGREGATION._serialized_start=1110
-  _SECUREAGGREGATION._serialized_end=1270
-  _SECUREAGGREGATION_NAMEDVALUESENTRY._serialized_start=1201
-  _SECUREAGGREGATION_NAMEDVALUESENTRY._serialized_end=1270
+  _TASKINS._serialized_end=501
+  _TASKRES._serialized_start=503
+  _TASKRES._serialized_end=595
+  _VALUE._serialized_start=598
+  _VALUE._serialized_end=1097
+  _VALUE_DOUBLELIST._serialized_start=953
+  _VALUE_DOUBLELIST._serialized_end=979
+  _VALUE_SINT64LIST._serialized_start=981
+  _VALUE_SINT64LIST._serialized_end=1007
+  _VALUE_BOOLLIST._serialized_start=1009
+  _VALUE_BOOLLIST._serialized_end=1033
+  _VALUE_STRINGLIST._serialized_start=1035
+  _VALUE_STRINGLIST._serialized_end=1061
+  _VALUE_BYTESLIST._serialized_start=1063
+  _VALUE_BYTESLIST._serialized_end=1088
+  _SECUREAGGREGATION._serialized_start=1100
+  _SECUREAGGREGATION._serialized_end=1260
+  _SECUREAGGREGATION_NAMEDVALUESENTRY._serialized_start=1191
+  _SECUREAGGREGATION_NAMEDVALUESENTRY._serialized_end=1260
 # @@protoc_insertion_point(module_scope)

flwr/proto/task_pb2.pyi

@@ -59,44 +59,44 @@ class TaskIns(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
     TASK_ID_FIELD_NUMBER: builtins.int
     GROUP_ID_FIELD_NUMBER: builtins.int
-    WORKLOAD_ID_FIELD_NUMBER: builtins.int
+    RUN_ID_FIELD_NUMBER: builtins.int
     TASK_FIELD_NUMBER: builtins.int
     task_id: typing.Text
     group_id: typing.Text
-    workload_id: builtins.int
+    run_id: builtins.int
     @property
     def task(self) -> global___Task: ...
     def __init__(self,
         *,
         task_id: typing.Text = ...,
         group_id: typing.Text = ...,
-        workload_id: builtins.int = ...,
+        run_id: builtins.int = ...,
         task: typing.Optional[global___Task] = ...,
         ) -> None: ...
     def HasField(self, field_name: typing_extensions.Literal["task",b"task"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["group_id",b"group_id","task",b"task","task_id",b"task_id","workload_id",b"workload_id"]) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["group_id",b"group_id","run_id",b"run_id","task",b"task","task_id",b"task_id"]) -> None: ...
 global___TaskIns = TaskIns
 
 class TaskRes(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
     TASK_ID_FIELD_NUMBER: builtins.int
     GROUP_ID_FIELD_NUMBER: builtins.int
-    WORKLOAD_ID_FIELD_NUMBER: builtins.int
+    RUN_ID_FIELD_NUMBER: builtins.int
     TASK_FIELD_NUMBER: builtins.int
     task_id: typing.Text
     group_id: typing.Text
-    workload_id: builtins.int
+    run_id: builtins.int
     @property
     def task(self) -> global___Task: ...
     def __init__(self,
         *,
         task_id: typing.Text = ...,
         group_id: typing.Text = ...,
-        workload_id: builtins.int = ...,
+        run_id: builtins.int = ...,
         task: typing.Optional[global___Task] = ...,
         ) -> None: ...
     def HasField(self, field_name: typing_extensions.Literal["task",b"task"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["group_id",b"group_id","task",b"task","task_id",b"task_id","workload_id",b"workload_id"]) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["group_id",b"group_id","run_id",b"run_id","task",b"task","task_id",b"task_id"]) -> None: ...
 global___TaskRes = TaskRes
 
 class Value(google.protobuf.message.Message):

flwr/server/driver/driver_servicer.py

@@ -51,7 +51,7 @@ class DriverServicer(driver_pb2_grpc.DriverServicer):
         """Get available nodes."""
         log(INFO, "DriverServicer.GetNodes")
         state: State = self.state_factory.state()
-        all_ids: Set[int] = state.get_nodes(request.workload_id)
+        all_ids: Set[int] = state.get_nodes(request.run_id)
         nodes: List[Node] = [
             Node(node_id=node_id, anonymous=False) for node_id in all_ids
         ]
@@ -63,8 +63,8 @@ class DriverServicer(driver_pb2_grpc.DriverServicer):
         """Create workload ID."""
         log(INFO, "DriverServicer.CreateWorkload")
         state: State = self.state_factory.state()
-        workload_id = state.create_workload()
-        return CreateWorkloadResponse(workload_id=workload_id)
+        run_id = state.create_workload()
+        return CreateWorkloadResponse(run_id=run_id)
 
     def PushTaskIns(
         self, request: PushTaskInsRequest, context: grpc.ServicerContext

flwr/server/state/in_memory_state.py

@@ -32,7 +32,7 @@ class InMemoryState(State):
 
     def __init__(self) -> None:
         self.node_ids: Set[int] = set()
-        self.workload_ids: Set[int] = set()
+        self.run_ids: Set[int] = set()
         self.task_ins_store: Dict[UUID, TaskIns] = {}
         self.task_res_store: Dict[UUID, TaskRes] = {}
 
@@ -43,9 +43,9 @@ class InMemoryState(State):
         if any(errors):
             log(ERROR, errors)
             return None
-        # Validate workload_id
-        if task_ins.workload_id not in self.workload_ids:
-            log(ERROR, "`workload_id` is invalid")
+        # Validate run_id
+        if task_ins.run_id not in self.run_ids:
+            log(ERROR, "`run_id` is invalid")
             return None
 
         # Create task_id, created_at and ttl
@@ -104,9 +104,9 @@ class InMemoryState(State):
             log(ERROR, errors)
             return None
 
-        # Validate workload_id
-        if task_res.workload_id not in self.workload_ids:
-            log(ERROR, "`workload_id` is invalid")
+        # Validate run_id
+        if task_res.run_id not in self.run_ids:
+            log(ERROR, "`run_id` is invalid")
             return None
 
         # Create task_id, created_at and ttl
@@ -199,25 +199,25 @@ class InMemoryState(State):
             raise ValueError(f"Node {node_id} not found")
         self.node_ids.remove(node_id)
 
-    def get_nodes(self, workload_id: int) -> Set[int]:
+    def get_nodes(self, run_id: int) -> Set[int]:
         """Return all available client nodes.
 
         Constraints
         -----------
-        If the provided `workload_id` does not exist or has no matching nodes,
+        If the provided `run_id` does not exist or has no matching nodes,
         an empty `Set` MUST be returned.
         """
-        if workload_id not in self.workload_ids:
+        if run_id not in self.run_ids:
             return set()
         return self.node_ids
 
     def create_workload(self) -> int:
         """Create one workload."""
-        # Sample a random int64 as workload_id
-        workload_id: int = int.from_bytes(os.urandom(8), "little", signed=True)
+        # Sample a random int64 as run_id
+        run_id: int = int.from_bytes(os.urandom(8), "little", signed=True)
 
-        if workload_id not in self.workload_ids:
-            self.workload_ids.add(workload_id)
-            return workload_id
+        if run_id not in self.run_ids:
+            self.run_ids.add(run_id)
+            return run_id
         log(ERROR, "Unexpected workload creation failure.")
         return 0

flwr/server/state/sqlite_state.py

@@ -39,7 +39,7 @@ CREATE TABLE IF NOT EXISTS node(
 
 SQL_CREATE_TABLE_WORKLOAD = """
 CREATE TABLE IF NOT EXISTS workload(
-    workload_id INTEGER UNIQUE
+    run_id INTEGER UNIQUE
 );
 """
 
@@ -47,7 +47,7 @@ SQL_CREATE_TABLE_TASK_INS = """
 CREATE TABLE IF NOT EXISTS task_ins(
     task_id                 TEXT UNIQUE,
     group_id                TEXT,
-    workload_id             INTEGER,
+    run_id             INTEGER,
     producer_anonymous      BOOLEAN,
     producer_node_id        INTEGER,
     consumer_anonymous      BOOLEAN,
@@ -58,7 +58,7 @@ CREATE TABLE IF NOT EXISTS task_ins(
     ancestry                TEXT,
     legacy_server_message   BLOB,
     legacy_client_message   BLOB,
-    FOREIGN KEY(workload_id) REFERENCES workload(workload_id)
+    FOREIGN KEY(run_id) REFERENCES workload(run_id)
 );
 """
 
@@ -67,7 +67,7 @@ SQL_CREATE_TABLE_TASK_RES = """
 CREATE TABLE IF NOT EXISTS task_res(
     task_id                 TEXT UNIQUE,
     group_id                TEXT,
-    workload_id             INTEGER,
+    run_id             INTEGER,
     producer_anonymous      BOOLEAN,
     producer_node_id        INTEGER,
     consumer_anonymous      BOOLEAN,
@@ -78,7 +78,7 @@ CREATE TABLE IF NOT EXISTS task_res(
     ancestry                TEXT,
     legacy_server_message   BLOB,
     legacy_client_message   BLOB,
-    FOREIGN KEY(workload_id) REFERENCES workload(workload_id)
+    FOREIGN KEY(run_id) REFERENCES workload(run_id)
 );
 """
 
@@ -198,7 +198,7 @@ class SqliteState(State):
         columns = ", ".join([f":{key}" for key in data[0]])
         query = f"INSERT INTO task_ins VALUES({columns});"
 
-        # Only invalid workload_id can trigger IntegrityError.
+        # Only invalid run_id can trigger IntegrityError.
         # This may need to be changed in the future version with more integrity checks.
         try:
             self.query(query, data)
@@ -333,7 +333,7 @@ class SqliteState(State):
         columns = ", ".join([f":{key}" for key in data[0]])
         query = f"INSERT INTO task_res VALUES({columns});"
 
-        # Only invalid workload_id can trigger IntegrityError.
+        # Only invalid run_id can trigger IntegrityError.
         # This may need to be changed in the future version with more integrity checks.
         try:
             self.query(query, data)
@@ -485,17 +485,17 @@ class SqliteState(State):
         query = "DELETE FROM node WHERE node_id = :node_id;"
         self.query(query, {"node_id": node_id})
 
-    def get_nodes(self, workload_id: int) -> Set[int]:
+    def get_nodes(self, run_id: int) -> Set[int]:
         """Retrieve all currently stored node IDs as a set.
 
         Constraints
         -----------
-        If the provided `workload_id` does not exist or has no matching nodes,
+        If the provided `run_id` does not exist or has no matching nodes,
         an empty `Set` MUST be returned.
         """
         # Validate workload ID
-        query = "SELECT COUNT(*) FROM workload WHERE workload_id = ?;"
-        if self.query(query, (workload_id,))[0]["COUNT(*)"] == 0:
+        query = "SELECT COUNT(*) FROM workload WHERE run_id = ?;"
+        if self.query(query, (run_id,))[0]["COUNT(*)"] == 0:
             return set()
 
         # Get nodes
@@ -506,16 +506,16 @@ class SqliteState(State):
 
     def create_workload(self) -> int:
         """Create one workload and store it in state."""
-        # Sample a random int64 as workload_id
-        workload_id: int = int.from_bytes(os.urandom(8), "little", signed=True)
+        # Sample a random int64 as run_id
+        run_id: int = int.from_bytes(os.urandom(8), "little", signed=True)
 
         # Check conflicts
-        query = "SELECT COUNT(*) FROM workload WHERE workload_id = ?;"
-        # If workload_id does not exist
-        if self.query(query, (workload_id,))[0]["COUNT(*)"] == 0:
-            query = "INSERT INTO workload VALUES(:workload_id);"
-            self.query(query, {"workload_id": workload_id})
-            return workload_id
+        query = "SELECT COUNT(*) FROM workload WHERE run_id = ?;"
+        # If run_id does not exist
+        if self.query(query, (run_id,))[0]["COUNT(*)"] == 0:
+            query = "INSERT INTO workload VALUES(:run_id);"
+            self.query(query, {"run_id": run_id})
+            return run_id
         log(ERROR, "Unexpected workload creation failure.")
         return 0
 
@@ -537,7 +537,7 @@ def task_ins_to_dict(task_msg: TaskIns) -> Dict[str, Any]:
     result = {
         "task_id": task_msg.task_id,
         "group_id": task_msg.group_id,
-        "workload_id": task_msg.workload_id,
+        "run_id": task_msg.run_id,
         "producer_anonymous": task_msg.task.producer.anonymous,
         "producer_node_id": task_msg.task.producer.node_id,
         "consumer_anonymous": task_msg.task.consumer.anonymous,
@@ -559,7 +559,7 @@ def task_res_to_dict(task_msg: TaskRes) -> Dict[str, Any]:
     result = {
         "task_id": task_msg.task_id,
         "group_id": task_msg.group_id,
-        "workload_id": task_msg.workload_id,
+        "run_id": task_msg.run_id,
         "producer_anonymous": task_msg.task.producer.anonymous,
         "producer_node_id": task_msg.task.producer.node_id,
         "consumer_anonymous": task_msg.task.consumer.anonymous,
@@ -584,7 +584,7 @@ def dict_to_task_ins(task_dict: Dict[str, Any]) -> TaskIns:
     result = TaskIns(
         task_id=task_dict["task_id"],
         group_id=task_dict["group_id"],
-        workload_id=task_dict["workload_id"],
+        run_id=task_dict["run_id"],
         task=Task(
             producer=Node(
                 node_id=task_dict["producer_node_id"],
@@ -612,7 +612,7 @@ def dict_to_task_res(task_dict: Dict[str, Any]) -> TaskRes:
     result = TaskRes(
         task_id=task_dict["task_id"],
         group_id=task_dict["group_id"],
-        workload_id=task_dict["workload_id"],
+        run_id=task_dict["run_id"],
         task=Task(
             producer=Node(
                 node_id=task_dict["producer_node_id"],

flwr/server/state/state.py

@@ -43,7 +43,7 @@ class State(abc.ABC):
         If `task_ins.task.consumer.anonymous` is `False`, then
         `task_ins.task.consumer.node_id` MUST be set (not 0)
 
-        If `task_ins.workload_id` is invalid, then
+        If `task_ins.run_id` is invalid, then
         storing the `task_ins` MUST fail.
         """
 
@@ -92,7 +92,7 @@ class State(abc.ABC):
         If `task_res.task.consumer.anonymous` is `False`, then
         `task_res.task.consumer.node_id` MUST be set (not 0)
 
-        If `task_res.workload_id` is invalid, then
+        If `task_res.run_id` is invalid, then
         storing the `task_res` MUST fail.
         """
 
@@ -140,12 +140,12 @@ class State(abc.ABC):
         """Remove `node_id` from state."""
 
     @abc.abstractmethod
-    def get_nodes(self, workload_id: int) -> Set[int]:
+    def get_nodes(self, run_id: int) -> Set[int]:
         """Retrieve all currently stored node IDs as a set.
 
         Constraints
         -----------
-        If the provided `workload_id` does not exist or has no matching nodes,
+        If the provided `run_id` does not exist or has no matching nodes,
         an empty `Set` MUST be returned.
         """
 

flwr/simulation/ray_transport/ray_client_proxy.py

@@ -133,15 +133,15 @@ class RayActorClientProxy(ClientProxy):
 
     def _submit_job(self, job_fn: JobFn, timeout: Optional[float]) -> ClientRes:
         # The VCE is not exposed to TaskIns, it won't handle multilple workloads
-        # For the time being, fixing workload_id is a small compromise
+        # For the time being, fixing run_id is a small compromise
         # This will be one of the first points to address integrating VCE + DriverAPI
-        workload_id = 0
+        run_id = 0
 
         # Register state
-        self.proxy_state.register_workloadstate(workload_id=workload_id)
+        self.proxy_state.register_workloadstate(run_id=run_id)
 
         # Retrieve state
-        state = self.proxy_state.retrieve_workloadstate(workload_id=workload_id)
+        state = self.proxy_state.retrieve_workloadstate(run_id=run_id)
 
         try:
             self.actor_pool.submit_client_job(
@@ -152,7 +152,7 @@ class RayActorClientProxy(ClientProxy):
 
             # Update state
             self.proxy_state.update_workloadstate(
-                workload_id=workload_id, workload_state=updated_state
+                run_id=run_id, workload_state=updated_state
             )
 
         except Exception as ex:

{flwr_nightly-1.7.0.dev20240103.dist-info → flwr_nightly-1.7.0.dev20240104.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: flwr-nightly
-Version: 1.7.0.dev20240103
+Version: 1.7.0.dev20240104
 Summary: Flower: A Friendly Federated Learning Framework
 Home-page: https://flower.dev
 License: Apache-2.0

{flwr_nightly-1.7.0.dev20240103.dist-info → flwr_nightly-1.7.0.dev20240104.dist-info}/RECORD

@@ -1,20 +1,20 @@
 flwr/__init__.py,sha256=6zbcS7z2q-VUdmpFppLH6BacsE-ZFmfq6OvtKNOyYE0,981
 flwr/client/__init__.py,sha256=2T4enmlE4PsoKiGTvXwBKSlhOjZ7MXRy5oCGNf0UH9Y,1111
-flwr/client/app.py,sha256=X5OGCBtKtEj-JBk9tXJyE6Rfllpo6Nse46pkznzofdg,19252
+flwr/client/app.py,sha256=te-CJj1VYsm6XTPPkVrvF1Tyg_fC_URU0bqaqDyPUw0,19176
 flwr/client/client.py,sha256=o7uPXVvLJExyqGz8W1JdjnVL3HTxvyAZ6hhIQTEjwrg,8233
 flwr/client/dpfedavg_numpy_client.py,sha256=LxUcPBO0mU3VScAx9vx2PlsghXjQZVEBOn3rolgrgio,7216
 flwr/client/flower.py,sha256=PPYROFVnu7NEe7sR8XFVbkfvfwecIuQc5-VllPz0LlQ,4049
 flwr/client/grpc_client/__init__.py,sha256=LsnbqXiJhgQcB0XzAlUQgPx011Uf7Y7yabIC1HxivJ8,735
-flwr/client/grpc_client/connection.py,sha256=5EcdnedH0SvNA4daIfRWzQLS_3a59SJb2o8RGzvnDO0,5007
+flwr/client/grpc_client/connection.py,sha256=ki7ZaYvSDJK1bvW2UxrldyToiEvqk2j_cuIaICT19ZA,5002
 flwr/client/grpc_rere_client/__init__.py,sha256=avn6W_vHEM_yZEB1S7hCZgnTbXb6ZujqRP_vAzyXu-0,752
 flwr/client/grpc_rere_client/connection.py,sha256=r0wzwJp9ugcHlD0iF6wj45KqpfDe_AwBbI9jUjiIVrI,6560
 flwr/client/message_handler/__init__.py,sha256=abHvBRJJiiaAMNgeILQbMOa6h8WqMK2BcnvxwQZFpic,719
-flwr/client/message_handler/message_handler.py,sha256=kX5ruNg71mqkFWC6zhBw_xFN3y7LwotoaIxZPOdOfhc,8360
-flwr/client/message_handler/task_handler.py,sha256=1ZTFP3nJejrIo_hEwhyjMpxf7jewI5kVKdHE5ljtZNY,5743
+flwr/client/message_handler/message_handler.py,sha256=DUUBNEFhRAby8Oyf0_M9O2ZgNSdnfEygHRd8RgUg-Gs,8355
+flwr/client/message_handler/task_handler.py,sha256=zy9BuJ_BLiF3-ngDlTA_cOZW6QfqNkYV00X4P7po9hE,5713
 flwr/client/middleware/__init__.py,sha256=Eo3JvAV5XqmyRySNqeiw93YNETmmP5ixEOMeBA6ah4w,769
 flwr/client/middleware/utils.py,sha256=QUghso_SWsKTUPfKwrtBwPyyJoEI9AV9hRY2acu1TYE,1168
-flwr/client/node_state.py,sha256=ZXL6L4GUku1X-lLcS-khiDwZTphGHdzUzj82vvopweQ,1933
-flwr/client/node_state_tests.py,sha256=-iHTtcjQJwo_D4gmiEnRtxR8Oz_7wRYzKQo8Xk8TwSU,1988
+flwr/client/node_state.py,sha256=3y6_Ibn_7Zg9KrkADPLwNBrl9dSJsC3KYC7VqecMw2U,1864
+flwr/client/node_state_tests.py,sha256=OBowSSvTg3ckHzGFmGpJnhrmJB0eyNJjv48fOm1_cmo,1963
 flwr/client/numpy_client.py,sha256=W1u85DLrT5s-gmjGI2SeT-OQyPBshHJOQwqT8OZErrk,10321
 flwr/client/rest_client/__init__.py,sha256=ThwOnkMdzxo_UuyTI47Q7y9oSpuTgNT2OuFvJCfuDiw,735
 flwr/client/rest_client/connection.py,sha256=0vzOKxdnqgLtne4kIXVREtMWr40HqlqIvE6ZRur-ViM,11583
@@ -45,14 +45,14 @@ flwr/common/telemetry.py,sha256=se_-pHgEWcmN09ChSpTeek72l1UJHf7GbwXBB1KXBjQ,7683
 flwr/common/typing.py,sha256=e21UHUqHIzucAkTt2Xh7sOO-qTCZjqEbY09lHaCE9tc,3959
 flwr/common/version.py,sha256=A0MKvyKPrV8wLg0YCAODTqM71v26NEH36c6JYtfgg0o,667
 flwr/driver/__init__.py,sha256=NQ4KeZ5fP9wdxGjcr2cP41_7TLuuYQ3u4J7GwYtQ488,870
-flwr/driver/app.py,sha256=h_Er4UDwdcX1agiHzl8sQtFmgSVLUkc3qxLUutLPTxA,7263
-flwr/driver/driver.py,sha256=zNeGvaq-t_JOeMLdN2UGsffPqNQ6dVUX6HpJPlLQbIs,3992
-flwr/driver/driver_client_proxy.py,sha256=qZznLAwy_wj_g_0hG74qxB-j0zniAiB16Y_y2ueqv7Q,5758
+flwr/driver/app.py,sha256=Lg21jwMTUyDjdA55n--ySNh-DK4se_YZou3Gditk1-g,7228
+flwr/driver/driver.py,sha256=1X8caHcchoeV96cKlGr9b8EZhWdJqTpzTjv6-6J7CyI,3912
+flwr/driver/driver_client_proxy.py,sha256=4OQWjX-L8FOfCStij9dzTqkw_lrkbHqNGDrLFiUFcbo,5719
 flwr/driver/grpc_driver.py,sha256=A9llpjE8diK_-s0vZ0yKnSRa1xr7y51d5JB8vfBbmOM,4526
 flwr/flower/__init__.py,sha256=VlOKQzq4OecSM4Ga2uk1iu2K8mxMc-j1XA0bdhaTeMw,892
 flwr/proto/__init__.py,sha256=hbY7JYakwZwCkYgCNlmHdc8rtvfoJbAZLalMdc--CGc,683
-flwr/proto/driver_pb2.py,sha256=UcS3znnj9lb0nTbPUgO8G43vJ4oOTXzp85AsxQhMTDw,5954
-flwr/proto/driver_pb2.pyi,sha256=-hpthCr8T4NAzHwPCBgPmw1-iySQu22eHSICxXGE2LA,4755
+flwr/proto/driver_pb2.py,sha256=kik2QjFAb9hNyrjMv_xMsThBzpsw5QxZSkzaur3PvwQ,5944
+flwr/proto/driver_pb2.pyi,sha256=o95CkfxURsu6rsIlfSuq8F4YSTHdPtDOPAgpw3GW19s,4705
 flwr/proto/driver_pb2_grpc.py,sha256=FL875wzg0E9qd70ssV55YpEwAdEHUt7XTu7MNL0soTQ,7374
 flwr/proto/driver_pb2_grpc.pyi,sha256=70HgehRecB93VgQQzv1goESXPHQeEdmb6A1St2lrpG0,2062
 flwr/proto/fleet_pb2.py,sha256=h_lDZDTEauPy53lomiYaSngwmveNkMw94rvCQjg3yx8,7295
@@ -63,8 +63,8 @@ flwr/proto/node_pb2.py,sha256=bWqn3w6wvUjBrQQFA4P5lIpfFXQQ4u0kEMavk4xqoQo,1188
 flwr/proto/node_pb2.pyi,sha256=aX3BHhgXvJE1rvcRnEE_gB-5GcaFQ0SJ88yTE223bjI,751
 flwr/proto/node_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
 flwr/proto/node_pb2_grpc.pyi,sha256=ff2TSiLVnG6IVQcTGzb2DIH3XRSoAvAo_RMcvbMFyc0,76
-flwr/proto/task_pb2.py,sha256=1aJ15GzR7Stvf6O7DMdxxcBflUeg2r-5chB5QLqAqNo,8236
-flwr/proto/task_pb2.pyi,sha256=Je30RzZFzUXhMGCrYPEHUwXNTcEUDpOYfA59AyWABaA,10893
+flwr/proto/task_pb2.py,sha256=k64g-fJh-PgKghjMKlDLWJcrlOH5aFswyzd6OzClhGs,8228
+flwr/proto/task_pb2.pyi,sha256=vdsSlvMkg9vN-qQsf6HgV_JbbYrBK17f-fkeJo8a55Y,10843
 flwr/proto/task_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
 flwr/proto/task_pb2_grpc.pyi,sha256=ff2TSiLVnG6IVQcTGzb2DIH3XRSoAvAo_RMcvbMFyc0,76
 flwr/proto/transport_pb2.py,sha256=XWXbVa9pylkn-1mdlSaV9x0lk8zIC40DZzBcB_3LaMk,18721
@@ -78,7 +78,7 @@ flwr/server/client_manager.py,sha256=T8UDSRJBVD3fyIDI7NTAA-NA7GPrMNNgH2OAF54RRxE
 flwr/server/client_proxy.py,sha256=8ScGDvP3jHbl8DV3hyFID5N5VEVlXn8ZTQXtkdOfssI,2234
 flwr/server/criterion.py,sha256=ypbAexbztzGUxNen9RCHF91QeqiEQix4t4Ih3E-42MM,1061
 flwr/server/driver/__init__.py,sha256=STB1_DASVEg7Cu6L7VYxTzV7UMkgtBkFim09Z82Dh8I,712
-flwr/server/driver/driver_servicer.py,sha256=Ru6iR2oIgVJmir48a_DXv0ph4R5P-k7l_CaK7iQHXdA,4518
+flwr/server/driver/driver_servicer.py,sha256=C_gPF4nHyBM3p1CNSDpBOb6GLqkuBeLocB81giL3hAU,4498
 flwr/server/fleet/__init__.py,sha256=C6GCSD5eP5Of6_dIeSe1jx9HnV0icsvWyQ5EKAUHJRU,711
 flwr/server/fleet/grpc_bidi/__init__.py,sha256=mgGJGjwT6VU7ovC1gdnnqttjyBPlNIcZnYRqx4K3IBQ,735
 flwr/server/fleet/grpc_bidi/driver_client_manager.py,sha256=179z_IVT_95G7dIMsP9ec0JpoPEA62Eu12YyM-TKsyM,4284
@@ -96,9 +96,9 @@ flwr/server/fleet/rest_rere/rest_api.py,sha256=_72gqtux9js0NBS-qnhLF--Pb93Em6sGP
 flwr/server/history.py,sha256=W7PHCFX7dLXrdnaVfl5V4tuzmtxh6zArkWYxVXvTZ1c,4904
 flwr/server/server.py,sha256=skrNgQp9vlCVHruovlaB0Rh1W7xdH7KqEfVCMZGpK7c,15965
 flwr/server/state/__init__.py,sha256=ij-7Ms-hyordQdRmGQxY1-nVa4OhixJ0jr7_YDkys0s,1003
-flwr/server/state/in_memory_state.py,sha256=j4redg12x5ULd40V-NLIuw5A1LtZZJo3xgzDINe0W3g,7983
-flwr/server/state/sqlite_state.py,sha256=wb0vSGUXN_A60I3qHj3bhosPxiPhbv1wRzRSqy-a-sI,21423
-flwr/server/state/state.py,sha256=gsCYc7c5OO3Y2O5k7McS2WIJNzvlswMTSpuTPIr20-E,5321
+flwr/server/state/in_memory_state.py,sha256=iDdk5y0vNkoiWPbEKj72JARzEP7pahWQ9Q_sQ-9eZp8,7883
+flwr/server/state/sqlite_state.py,sha256=myd9MsI3aeDDBJG3oyDDV4F5q9aJCOI4jrtol40lAr0,21273
+flwr/server/state/state.py,sha256=d464gfbAdBpqpcKZ8MlKphJruE4MX-x-8G5ENg1UB4M,5301
 flwr/server/state/state_factory.py,sha256=91cSB-KOAFM37z7T098WxTkVeKNaAZ_mTI75snn2_tk,1654
 flwr/server/strategy/__init__.py,sha256=EDTv_lU67VmZ8pRqy5fabQDhq5x4oRiD-KHoXhOIWMs,2096
 flwr/server/strategy/aggregate.py,sha256=KhxPZbc0riuxci214jNDsiD65jT7XLW6oaU2Gux9u9s,12528
@@ -129,10 +129,10 @@ flwr/simulation/__init__.py,sha256=E2eD5FlTmZZ80u21FmWCkacrM7O4mrEHD8iXqeCaBUQ,1
 flwr/simulation/app.py,sha256=ep8QOHPGCZ3121eu9EJ0T9xnM_KgTyDjYKRakgQ7_F4,13098
 flwr/simulation/ray_transport/__init__.py,sha256=FsaAnzC4cw4DqoouBCix6496k29jACkfeIam55BvW9g,734
 flwr/simulation/ray_transport/ray_actor.py,sha256=kCaH8I-F7Ckk1LbwTSpYvwB8Rqar8Kl1V8jbhwmHcQg,17133
-flwr/simulation/ray_transport/ray_client_proxy.py,sha256=FIyqnhBU2I34ZAHRJrPIRYoNQwz1PaU7c0ERiHSepDE,9530
+flwr/simulation/ray_transport/ray_client_proxy.py,sha256=u_tvBCY3EiL7FAbWJwveQHdqLBrp6T2d7i-HZd_ftdo,9490
 flwr/simulation/ray_transport/utils.py,sha256=v36Oy9U4adcVjrqGUJjoWNnZVd_KFoixLmRmDe_CCzs,3373
-flwr_nightly-1.7.0.dev20240103.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-flwr_nightly-1.7.0.dev20240103.dist-info/METADATA,sha256=Wrv5Il7WXbD9nUCTMoLsuZuq4MGh55ApRTDacZAAMyo,13449
-flwr_nightly-1.7.0.dev20240103.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-flwr_nightly-1.7.0.dev20240103.dist-info/entry_points.txt,sha256=1uLlD5tIunkzALMfMWnqjdE_D5hRUX_I1iMmOMv6tZI,181
-flwr_nightly-1.7.0.dev20240103.dist-info/RECORD,,
+flwr_nightly-1.7.0.dev20240104.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+flwr_nightly-1.7.0.dev20240104.dist-info/METADATA,sha256=H_d6qitb63LZRKdoUoK5Ayn40-ac7D_bViwn0uO87bA,13449
+flwr_nightly-1.7.0.dev20240104.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+flwr_nightly-1.7.0.dev20240104.dist-info/entry_points.txt,sha256=1uLlD5tIunkzALMfMWnqjdE_D5hRUX_I1iMmOMv6tZI,181
+flwr_nightly-1.7.0.dev20240104.dist-info/RECORD,,