Adding changes

Signed-off-by: Agrawal, Kush <[email protected]>
securefederatedai · Feb 21, 2025 · 91594d3 · 91594d3
1 parent 22f5288
commit 91594d3
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/openfl/protocols/utils.py b/openfl/protocols/utils.py
@@ -2,6 +2,7 @@
 # SPDX-License-Identifier: Apache-2.0
 
 """Proto utils."""
+
 from openfl.protocols import base_pb2
 from openfl.utilities import TensorKey
 
@@ -345,9 +346,11 @@ def proto_to_datastream(proto, logger, max_buffer_size=(2 * 1024 * 1024)):
         yield reply
         reply = None
         chunk = None
+        del reply, chunk
 
     del npbytes, npbytes_view
 
+
 def get_headers(context) -> dict:
     """Get headers from context.
 

diff --git a/openfl/transport/grpc/aggregator_client.py b/openfl/transport/grpc/aggregator_client.py
@@ -466,8 +466,9 @@ def send_local_task_results(
         )
 
         # convert (potentially) long list of tensors into stream
-        stream = []
-        stream += utils.proto_to_datastream(request, self.logger)
+        # stream = []
+        # stream += utils.proto_to_datastream(request, self.logger)
+        stream = utils.proto_to_datastream(request, self.logger)
         response = self.stub.SendLocalTaskResults(iter(stream))
 
         # also do other validation, like on the round_number