wip

Author: yakov-g

custom_model_runner/datarobot_drum/drum/gunicorn/gunicorn.conf.py

@@ -66,7 +66,7 @@
     if temp_loglevel in {"debug", "info", "warning", "error", "critical"}:
         loglevel = temp_loglevel
 
-bind = os.environ.get("ADDRESS", "0.0.0.0:8080")
+bind = os.environ["ADDRESS"]
 # loglevel = "info"
 accesslog = "-"
 access_log_format = '%(h)s %(l)s %(u)s %(t)s "%(r)s" %(s)s %(b)s "%(f)s" "%(a)s"'

custom_model_runner/datarobot_drum/drum/utils/setup.py

@@ -5,6 +5,7 @@
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
 
+import os
 import sys
 
 from datarobot_drum import RuntimeParameters
@@ -48,6 +49,10 @@ def setup_options(args=None):
 
     options = arg_parser.parse_args(args)
 
+    # Set ADDRESS env var from parsed options if not already set
+    if getattr(options, 'address', None) and 'ADDRESS' not in os.environ:
+        os.environ['ADDRESS'] = options.address
+
     """Set max workers from runtime parameters if available."""
     if RuntimeParameters.has("CUSTOM_MODEL_WORKERS"):
         options.max_workers = RuntimeParameters.get("CUSTOM_MODEL_WORKERS")