Skip to content

Commit 8e57d4a

Browse files
MasterSkepticistaishaileshpant
authored andcommitted
Update RPC docstring
Signed-off-by: Shah, Karan <kbshah1998@outlook.com>
1 parent 7324fef commit 8e57d4a

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

openfl/transport/grpc/aggregator_client.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,13 +6,14 @@
66

77
import logging
88
import time
9-
from typing import Optional, Tuple
9+
from typing import List, Optional, Tuple
1010

1111
import grpc
1212

1313
from openfl.protocols import aggregator_pb2, aggregator_pb2_grpc, utils
1414
from openfl.protocols.aggregator_client_interface import AggregatorClientInterface
1515
from openfl.protocols import aggregator_pb2, aggregator_pb2_grpc
16+
from openfl.protocols import aggregator_pb2, aggregator_pb2_grpc, base_pb2, utils
1617
from openfl.transport.grpc.common import create_header, create_insecure_channel, create_tls_channel
1718

1819
logger = logging.getLogger(__name__)
@@ -349,7 +350,7 @@ def get_aggregated_tensors(
349350
self,
350351
tensor_keys,
351352
require_lossless,
352-
):
353+
) -> List[base_pb2.NamedTensor]:
353354
"""
354355
Get aggregated tensors from the aggregator.
355356
@@ -358,7 +359,7 @@ def get_aggregated_tensors(
358359
require_lossless (bool): Whether lossless compression is required.
359360
360361
Returns:
361-
aggregator_pb2.TensorProto: The aggregated tensor.
362+
A list of `NamedTensor`s in the same order as requested.
362363
"""
363364
header = create_header(
364365
sender=self.collaborator_name,

0 commit comments

Comments
 (0)