1616logger = logging .getLogger (__name__ )
1717
1818PRED_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+ )
2022
2123
2224def get_rest_microservice (user_model , seldon_metrics ):
@@ -124,7 +126,9 @@ def HealthPing():
124126 @app .route ("/health/status" , methods = ["GET" ])
125127 def HealthStatus ():
126128 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+ )
128132 logger .debug ("REST Health Status Response: %s" , response )
129133 return jsonify (response )
130134
@@ -220,17 +224,23 @@ def Aggregate(self, request_grpc, context):
220224 )
221225
222226
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+ ):
224230 seldon_model = SeldonModelGRPC (user_model , seldon_metrics )
225231 options = []
226232 if ANNOTATION_GRPC_MAX_MSG_SIZE in annotations :
227233 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+ )
229237 options .append (("grpc.max_message_length" , max_msg ))
230238 options .append (("grpc.max_send_message_length" , max_msg ))
231239 options .append (("grpc.max_receive_message_length" , max_msg ))
232240
233- server = grpc .server (futures .ThreadPoolExecutor (max_workers = 10 ), options = options )
241+ server = grpc .server (
242+ futures .ThreadPoolExecutor (max_workers = 10 ), options = options
243+ )
234244
235245 if trace_interceptor :
236246 from grpc_opentracing .grpcext import intercept_server
@@ -240,7 +250,9 @@ def get_grpc_server(user_model, seldon_metrics, annotations={}, trace_intercepto
240250 prediction_pb2_grpc .add_GenericServicer_to_server (seldon_model , server )
241251 prediction_pb2_grpc .add_ModelServicer_to_server (seldon_model , server )
242252 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+ )
244256 prediction_pb2_grpc .add_CombinerServicer_to_server (seldon_model , server )
245257 prediction_pb2_grpc .add_RouterServicer_to_server (seldon_model , server )
246258 prediction_pb2_grpc .add_SeldonServicer_to_server (seldon_model , server )
0 commit comments