16
16
logger = logging .getLogger (__name__ )
17
17
18
18
PRED_UNIT_ID = os .environ .get ("PREDICTIVE_UNIT_ID" , "0" )
19
- METRICS_ENDPOINT = os .environ .get ("PREDICTIVE_UNIT_METRICS_ENDPOINT" , "/metrics" )
19
+ METRICS_ENDPOINT = os .environ .get (
20
+ "PREDICTIVE_UNIT_METRICS_ENDPOINT" , "/metrics"
21
+ )
20
22
21
23
22
24
def get_rest_microservice (user_model , seldon_metrics ):
@@ -124,7 +126,9 @@ def HealthPing():
124
126
@app .route ("/health/status" , methods = ["GET" ])
125
127
def HealthStatus ():
126
128
logger .debug ("REST Health Status Request" )
127
- response = seldon_core .seldon_methods .health_status (user_model , seldon_metrics )
129
+ response = seldon_core .seldon_methods .health_status (
130
+ user_model , seldon_metrics
131
+ )
128
132
logger .debug ("REST Health Status Response: %s" , response )
129
133
return jsonify (response )
130
134
@@ -220,17 +224,23 @@ def Aggregate(self, request_grpc, context):
220
224
)
221
225
222
226
223
- def get_grpc_server (user_model , seldon_metrics , annotations = {}, trace_interceptor = None ):
227
+ def get_grpc_server (
228
+ user_model , seldon_metrics , annotations = {}, trace_interceptor = None
229
+ ):
224
230
seldon_model = SeldonModelGRPC (user_model , seldon_metrics )
225
231
options = []
226
232
if ANNOTATION_GRPC_MAX_MSG_SIZE in annotations :
227
233
max_msg = int (annotations [ANNOTATION_GRPC_MAX_MSG_SIZE ])
228
- logger .info ("Setting grpc max message and receive length to %d" , max_msg )
234
+ logger .info (
235
+ "Setting grpc max message and receive length to %d" , max_msg
236
+ )
229
237
options .append (("grpc.max_message_length" , max_msg ))
230
238
options .append (("grpc.max_send_message_length" , max_msg ))
231
239
options .append (("grpc.max_receive_message_length" , max_msg ))
232
240
233
- server = grpc .server (futures .ThreadPoolExecutor (max_workers = 10 ), options = options )
241
+ server = grpc .server (
242
+ futures .ThreadPoolExecutor (max_workers = 10 ), options = options
243
+ )
234
244
235
245
if trace_interceptor :
236
246
from grpc_opentracing .grpcext import intercept_server
@@ -240,7 +250,9 @@ def get_grpc_server(user_model, seldon_metrics, annotations={}, trace_intercepto
240
250
prediction_pb2_grpc .add_GenericServicer_to_server (seldon_model , server )
241
251
prediction_pb2_grpc .add_ModelServicer_to_server (seldon_model , server )
242
252
prediction_pb2_grpc .add_TransformerServicer_to_server (seldon_model , server )
243
- prediction_pb2_grpc .add_OutputTransformerServicer_to_server (seldon_model , server )
253
+ prediction_pb2_grpc .add_OutputTransformerServicer_to_server (
254
+ seldon_model , server
255
+ )
244
256
prediction_pb2_grpc .add_CombinerServicer_to_server (seldon_model , server )
245
257
prediction_pb2_grpc .add_RouterServicer_to_server (seldon_model , server )
246
258
prediction_pb2_grpc .add_SeldonServicer_to_server (seldon_model , server )
0 commit comments