| 569 | return resp |
| 570 | |
| 571 | def init_grpc_stub(self): |
| 572 | # https://github.com/tensorflow/serving/issues/1382 |
| 573 | options = [('grpc.max_receive_message_length', self.max_body_size), |
| 574 | ('grpc.max_send_message_length', self.max_body_size)] |
| 575 | |
| 576 | endpoints = [self.ip + ":" + self.server_port] |
| 577 | g_endpoint = 'ipv4:{}'.format(','.join(endpoints)) |
| 578 | self.channel_ = grpc.insecure_channel(g_endpoint, options=options) |
| 579 | self.stub_ = general_model_service_pb2_grpc.GeneralModelServiceStub( |
| 580 | self.channel_) |
| 581 | |
| 582 | def __del__(self): |
| 583 | self.requests_session.close() |