move envs to gunicorn

Author: yakov-g

custom_model_runner/datarobot_drum/drum/entry_point.py

@@ -3,11 +3,13 @@
 from datarobot_drum import RuntimeParameters
 from datarobot_drum.drum.utils.setup import setup_options
 
+from custom_model_runner.datarobot_drum.drum.enum import ArgumentsOptions
+
 
 def run_drum_server():
     options = setup_options()
     if (
-        options.subparser_name == "server"
+        options.subparser_name == ArgumentsOptions.SERVER
         and RuntimeParameters.has("DRUM_SERVER_TYPE")
         and str(RuntimeParameters.get("DRUM_SERVER_TYPE")).lower() == "gunicorn"
     ):

public_dropin_environments/java_codegen/Dockerfile

@@ -69,6 +69,8 @@ ENV ADDRESS=0.0.0.0:8080
 ENV WITH_ERROR_SERVER=0 \
     PYTHONUNBUFFERED=1
 
+# Gunicorn configuration
+ENV MLOPS_RUNTIME_PARAM_DRUM_SERVER_TYPE='{"type": "string", "payload": "gunicorn"}'
 
 COPY ./*.sh ${CODE_DIR}/
 WORKDIR ${CODE_DIR}

public_dropin_environments/python311/Dockerfile

@@ -59,6 +59,8 @@ ENV ADDRESS=0.0.0.0:8080
 ENV WITH_ERROR_SERVER=0 \
     PYTHONUNBUFFERED=1
 
+# Gunicorn configuration
+ENV MLOPS_RUNTIME_PARAM_DRUM_SERVER_TYPE='{"type": "string", "payload": "gunicorn"}'
 
 COPY ./*.sh ${CODE_DIR}/
 WORKDIR ${CODE_DIR}

public_dropin_environments/python3_keras/Dockerfile

@@ -59,6 +59,9 @@ ENV ADDRESS=0.0.0.0:8080
 ENV WITH_ERROR_SERVER=0 \
     PYTHONUNBUFFERED=1
 
+# Gunicorn configuration
+ENV MLOPS_RUNTIME_PARAM_DRUM_SERVER_TYPE='{"type": "string", "payload": "gunicorn"}'
+
 COPY ./*.sh ${CODE_DIR}/
 WORKDIR ${CODE_DIR}
 

public_dropin_environments/python3_onnx/Dockerfile

@@ -59,6 +59,8 @@ ENV ADDRESS=0.0.0.0:8080
 ENV WITH_ERROR_SERVER=0 \
     PYTHONUNBUFFERED=1
 
+# Gunicorn configuration
+ENV MLOPS_RUNTIME_PARAM_DRUM_SERVER_TYPE='{"type": "string", "payload": "gunicorn"}'
 
 COPY ./*.sh ${CODE_DIR}/
 WORKDIR ${CODE_DIR}

public_dropin_environments/python3_pytorch/Dockerfile

@@ -59,6 +59,8 @@ ENV ADDRESS=0.0.0.0:8080
 ENV WITH_ERROR_SERVER=0 \
     PYTHONUNBUFFERED=1
 
+# Gunicorn configuration
+ENV MLOPS_RUNTIME_PARAM_DRUM_SERVER_TYPE='{"type": "string", "payload": "gunicorn"}'
 
 COPY ./*.sh ${CODE_DIR}/
 WORKDIR ${CODE_DIR}

public_dropin_environments/python3_sklearn/Dockerfile

@@ -64,7 +64,6 @@ ENV WITH_ERROR_SERVER=0 \
 # Gunicorn configuration
 ENV MLOPS_RUNTIME_PARAM_DRUM_SERVER_TYPE='{"type": "string", "payload": "gunicorn"}'
 
-
 COPY ./*.sh ${CODE_DIR}/
 WORKDIR ${CODE_DIR}
 

public_dropin_environments/python3_xgboost/Dockerfile

@@ -66,6 +66,9 @@ ENV WITH_ERROR_SERVER=0 \
 # Maybe the same would apply to gunicorn, when we test.
 # ENV OMP_NUM_THREADS=1
 
+# Gunicorn configuration
+ENV MLOPS_RUNTIME_PARAM_DRUM_SERVER_TYPE='{"type": "string", "payload": "gunicorn"}'
+
 COPY ./*.sh ${CODE_DIR}/
 WORKDIR ${CODE_DIR}
 

public_dropin_environments/r_lang/Dockerfile

@@ -158,6 +158,9 @@ ENV ADDRESS=0.0.0.0:8080
 ENV WITH_ERROR_SERVER=0 \
     PYTHONUNBUFFERED=1
 
+# Gunicorn configuration
+ENV MLOPS_RUNTIME_PARAM_DRUM_SERVER_TYPE='{"type": "string", "payload": "gunicorn"}'
+
 COPY ./*.sh ${CODE_DIR}/
 WORKDIR ${CODE_DIR}