Skip to content

Commit d255b40

Browse files
author
Eric Meadows
committed
Removed setting
1 parent fea8b45 commit d255b40

File tree

1 file changed

+18
-6
lines changed

1 file changed

+18
-6
lines changed

python/seldon_core/wrapper.py

Lines changed: 18 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,9 @@
1616
logger = logging.getLogger(__name__)
1717

1818
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+
)
2022

2123

2224
def 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

Comments
 (0)