def add_AgentServicer_to_server()

in lab/app/agent_pb2_grpc.py [0:0]

• 41 lines of code
• 1 McCabe index (conditional complexity)

def add_AgentServicer_to_server(servicer, server):
    rpc_method_handlers = {
            'Predict': grpc.unary_unary_rpc_method_handler(
                    servicer.Predict,
                    request_deserializer=agent__pb2.PredictRequest.FromString,
                    response_serializer=agent__pb2.PredictResponse.SerializeToString,
            ),
            'LoadModel': grpc.unary_unary_rpc_method_handler(
                    servicer.LoadModel,
                    request_deserializer=agent__pb2.LoadModelRequest.FromString,
                    response_serializer=agent__pb2.LoadModelResponse.SerializeToString,
            ),
            'UnLoadModel': grpc.unary_unary_rpc_method_handler(
                    servicer.UnLoadModel,
                    request_deserializer=agent__pb2.UnLoadModelRequest.FromString,
                    response_serializer=agent__pb2.UnLoadModelResponse.SerializeToString,
            ),
            'ListModels': grpc.unary_unary_rpc_method_handler(
                    servicer.ListModels,
                    request_deserializer=agent__pb2.ListModelsRequest.FromString,
                    response_serializer=agent__pb2.ListModelsResponse.SerializeToString,
            ),
            'DescribeModel': grpc.unary_unary_rpc_method_handler(
                    servicer.DescribeModel,
                    request_deserializer=agent__pb2.DescribeModelRequest.FromString,
                    response_serializer=agent__pb2.DescribeModelResponse.SerializeToString,
            ),
            'CaptureData': grpc.unary_unary_rpc_method_handler(
                    servicer.CaptureData,
                    request_deserializer=agent__pb2.CaptureDataRequest.FromString,
                    response_serializer=agent__pb2.CaptureDataResponse.SerializeToString,
            ),
            'GetCaptureDataStatus': grpc.unary_unary_rpc_method_handler(
                    servicer.GetCaptureDataStatus,
                    request_deserializer=agent__pb2.GetCaptureDataStatusRequest.FromString,
                    response_serializer=agent__pb2.GetCaptureDataStatusResponse.SerializeToString,
            ),
    }
    generic_handler = grpc.method_handlers_generic_handler(
            'AWS.SageMaker.Edge.Agent', rpc_method_handlers)
    server.add_generic_rpc_handlers((generic_handler,))