added microservice

SeldonIO · Aug 7, 2019 · c3dbca6 · c3dbca6
1 parent 1894266
commit c3dbca6
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/python/seldon_core/microservice.py b/python/seldon_core/microservice.py
@@ -17,6 +17,7 @@
 
 PARAMETERS_ENV_NAME = "PREDICTIVE_UNIT_PARAMETERS"
 SERVICE_PORT_ENV_NAME = "PREDICTIVE_UNIT_SERVICE_PORT"
+LOG_LEVEL_ENV = "SELDON_LOG_LEVEL"
 DEFAULT_PORT = 5000
 
 DEBUG_PARAMETER = "SELDON_DEBUG"
@@ -174,7 +175,8 @@ def main():
     parameters = parse_parameters(json.loads(args.parameters))
 
     # set up log level
-    log_level_num = getattr(logging, args.log_level.upper(), None)
+    log_level_raw = os.environ.get(LOG_LEVEL_ENV, args.log_level.upper())
+    log_level_num = getattr(logging, log_level_raw, None)
     if not isinstance(log_level_num, int):
         raise ValueError('Invalid log level: %s', args.log_level)