Skip to content

[RAPTOR-14627] use gunicorn in sklearn env #1659

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
yakov-g wants to merge 4 commits into
base: master
Choose a base branch
from
Open

[RAPTOR-14627] use gunicorn in sklearn env #1659

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
88f8148
don't test with error server
yakov-g Dec 17, 2025
33fc6cf
move all the envs to gunicorn
yakov-g Dec 10, 2025
a86d8b7
bump drum in envs to v1.17.10
yakov-g Dec 18, 2025
5383bbf
Reconcile dependencies, updated IDs, tags
Dec 18, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions public_dropin_environments/java_codegen/Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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}
Expand Down
6 changes: 3 additions & 3 deletions public_dropin_environments/java_codegen/env_info.json
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
"description": "This template can be used as an environment for DataRobot generated scoring code or models that implement the either the IClassificationPredictor or IRegressionPredictor interface from the datarobot-prediction package and for H2O models exported as POJO or MOJO.",
"programmingLanguage": "java",
"label": "",
"environmentVersionId": "693734d134f605124465ef0a",
"environmentVersionId": "69447f109bcfd9076dcb0127",
"environmentVersionDescription": "",
"isPublic": true,
"isDownloadable": true,
Expand All @@ -14,8 +14,8 @@
"contextUrl": "https://github.com/datarobot/datarobot-user-models/tree/master/public_dropin_environments/java_codegen",
"imageRepository": "env-java-codegen",
"tags": [
"v11.4.0-693734d134f605124465ef0a",
"693734d134f605124465ef0a",
"v11.4.0-69447f109bcfd9076dcb0127",
"69447f109bcfd9076dcb0127",
"v11.4.0-latest"
]
}
6 changes: 3 additions & 3 deletions public_dropin_environments/java_codegen/requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
#
# This file is autogenerated by pip-compile with Python 3.10
# This file is autogenerated by pip-compile with Python 3.11
# by the following command:
#
# pip-compile --index-url=https://pypi.org/simple --no-annotate --no-emit-index-url --no-emit-trusted-host --output-file=requirements.txt requirements.in
# pip-compile --cert=None --client-cert=None --index-url=https://pypi.org/simple --no-annotate --no-emit-index-url --no-emit-trusted-host --output-file=requirements.txt --pip-args=None requirements.in
#
annotated-types==0.7.0
argcomplete==3.6.3
Expand All @@ -19,7 +19,7 @@ charset-normalizer==3.4.4
click==8.3.1
cryptography==46.0.3
datarobot==3.10.0
datarobot-drum==1.17.8
datarobot-drum==1.17.10
datarobot-mlops==11.1.0
datarobot-storage==2.2.0
docker==7.1.0
Expand Down
2 changes: 2 additions & 0 deletions public_dropin_environments/python311/Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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}
Expand Down
6 changes: 3 additions & 3 deletions public_dropin_environments/python311/env_info.json
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
"description": "This template environment can be used to create Python based custom models. User is responsible to provide requirements.txt with the model, to install all the required dependencies.",
"programmingLanguage": "python",
"label": "",
"environmentVersionId": "693734dc34f605126d279d35",
"environmentVersionId": "69447f1e9bcfd90793c17021",
"environmentVersionDescription": "",
"isPublic": true,
"isDownloadable": true,
Expand All @@ -14,8 +14,8 @@
"contextUrl": "https://github.com/datarobot/datarobot-user-models/tree/master/public_dropin_environments/python311",
"imageRepository": "env-python",
"tags": [
"v11.4.0-693734dc34f605126d279d35",
"693734dc34f605126d279d35",
"v11.4.0-69447f1e9bcfd90793c17021",
"69447f1e9bcfd90793c17021",
"v11.4.0-latest"
]
}
6 changes: 3 additions & 3 deletions public_dropin_environments/python311/requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
#
# This file is autogenerated by pip-compile with Python 3.10
# This file is autogenerated by pip-compile with Python 3.11
# by the following command:
#
# pip-compile --index-url=https://pypi.org/simple --no-annotate --no-emit-index-url --no-emit-trusted-host --output-file=requirements.txt requirements.in
# pip-compile --cert=None --client-cert=None --index-url=https://pypi.org/simple --no-annotate --no-emit-index-url --no-emit-trusted-host --output-file=requirements.txt --pip-args=None requirements.in
#
annotated-types==0.7.0
argcomplete==3.6.3
Expand All @@ -19,7 +19,7 @@ charset-normalizer==3.4.4
click==8.3.1
cryptography==46.0.3
datarobot==3.10.0
datarobot-drum==1.17.8
datarobot-drum==1.17.10
datarobot-mlops==11.1.0
datarobot-storage==2.2.0
docker==7.1.0
Expand Down
6 changes: 3 additions & 3 deletions public_dropin_environments/python311_genai_agents/env_info.json
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
"description": "This template environment can be used to create GenAI-powered agents using CrewAI, LangGraph, or Llama-Index. Similar to other drop-in environments, you can either include a .pth artifact or any other code needed to deserialize your model, and optionally a custom.py file. You can also use this environment in codespaces.",
"programmingLanguage": "python",
"label": "",
"environmentVersionId": "69417fa8b1ce07076dca7611",
"environmentVersionId": "69447f269bcfd907abb4fadf",
"environmentVersionDescription": "",
"isPublic": true,
"isDownloadable": true,
Expand All @@ -15,8 +15,8 @@
"contextUrl": "https://github.com/datarobot/datarobot-user-models/tree/master/public_dropin_environments/python311_genai_agents",
"imageRepository": "env-python-genai-agents",
"tags": [
"v11.4.0-69417fa8b1ce07076dca7611",
"69417fa8b1ce07076dca7611",
"v11.4.0-69447f269bcfd907abb4fadf",
"69447f269bcfd907abb4fadf",
"v11.4.0-latest"
]
}
2 changes: 1 addition & 1 deletion public_dropin_environments/python311_genai_agents/requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
datarobot-drum==1.17.8
datarobot-drum==1.17.10
datarobot-genai==0.2.7
datarobot-mlops==11.1.0
datarobot-moderations==11.2.8
Expand Down
3 changes: 3 additions & 0 deletions public_dropin_environments/python3_keras/Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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}

Expand Down
6 changes: 3 additions & 3 deletions public_dropin_environments/python3_keras/env_info.json
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
"description": "This template environment can be used to create artifact-only keras custom models. This environment contains keras backed by tensorflow and only requires your model artifact as a .h5 file and optionally a custom.py file.",
"programmingLanguage": "python",
"label": "",
"environmentVersionId": "693734e634f60512884447a9",
"environmentVersionId": "69447f279bcfd907beabbacc",
"environmentVersionDescription": "",
"isPublic": true,
"isDownloadable": true,
Expand All @@ -14,8 +14,8 @@
"contextUrl": "https://github.com/datarobot/datarobot-user-models/tree/master/public_dropin_environments/python3_keras",
"imageRepository": "env-python-keras",
"tags": [
"v11.4.0-693734e634f60512884447a9",
"693734e634f60512884447a9",
"v11.4.0-69447f279bcfd907beabbacc",
"69447f279bcfd907beabbacc",
"v11.4.0-latest"
]
}
6 changes: 3 additions & 3 deletions public_dropin_environments/python3_keras/requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
#
# This file is autogenerated by pip-compile with Python 3.10
# This file is autogenerated by pip-compile with Python 3.11
# by the following command:
#
# pip-compile --index-url=https://pypi.org/simple --no-annotate --no-emit-index-url --no-emit-trusted-host --output-file=requirements.txt requirements.in
# pip-compile --cert=None --client-cert=None --index-url=https://pypi.org/simple --no-annotate --no-emit-index-url --no-emit-trusted-host --output-file=requirements.txt --pip-args=None requirements.in
#
absl-py==2.3.1
annotated-types==0.7.0
Expand All @@ -21,7 +21,7 @@ charset-normalizer==3.4.4
click==8.3.1
cryptography==46.0.3
datarobot==3.10.0
datarobot-drum==1.17.8
datarobot-drum==1.17.10
datarobot-mlops==11.1.0
datarobot-storage==2.2.0
docker==7.1.0
Expand Down
2 changes: 2 additions & 0 deletions public_dropin_environments/python3_onnx/Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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}
Expand Down
6 changes: 3 additions & 3 deletions public_dropin_environments/python3_onnx/env_info.json
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
"description": "This template environment can be used to create artifact-only ONNX custom models. This environment contains ONNX runtime and only requires your model artifact as an .onnx file and optionally a custom.py file.",
"programmingLanguage": "python",
"label": "",
"environmentVersionId": "693734f634f60512aea4ca03",
"environmentVersionId": "69447f3a9bcfd907e14f4fdc",
"environmentVersionDescription": "",
"isPublic": true,
"isDownloadable": true,
Expand All @@ -14,8 +14,8 @@
"contextUrl": "https://github.com/datarobot/datarobot-user-models/tree/master/public_dropin_environments/python3_onnx",
"imageRepository": "env-python-onnx",
"tags": [
"v11.4.0-693734f634f60512aea4ca03",
"693734f634f60512aea4ca03",
"v11.4.0-69447f3a9bcfd907e14f4fdc",
"69447f3a9bcfd907e14f4fdc",
"v11.4.0-latest"
]
}
6 changes: 3 additions & 3 deletions public_dropin_environments/python3_onnx/requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
#
# This file is autogenerated by pip-compile with Python 3.10
# This file is autogenerated by pip-compile with Python 3.11
# by the following command:
#
# pip-compile --index-url=https://pypi.org/simple --no-annotate --no-emit-index-url --no-emit-trusted-host --output-file=requirements.txt requirements.in
# pip-compile --cert=None --client-cert=None --index-url=https://pypi.org/simple --no-annotate --no-emit-index-url --no-emit-trusted-host --output-file=requirements.txt --pip-args=None requirements.in
#
annotated-types==0.7.0
argcomplete==3.6.3
Expand All @@ -20,7 +20,7 @@ click==8.3.1
coloredlogs==15.0.1
cryptography==46.0.3
datarobot==3.10.0
datarobot-drum==1.17.8
datarobot-drum==1.17.10
datarobot-mlops==11.1.0
datarobot-storage==2.2.0
docker==7.1.0
Expand Down
2 changes: 2 additions & 0 deletions public_dropin_environments/python3_pytorch/Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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}
Expand Down
6 changes: 3 additions & 3 deletions public_dropin_environments/python3_pytorch/env_info.json
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
"description": "This template environment can be used to create artifact-only PyTorch custom models. This environment contains PyTorch and requires only your model artifact as a .pth file, any other code needed to deserialize your model, and optionally a custom.py file.",
"programmingLanguage": "python",
"label": "",
"environmentVersionId": "6937350034f60512cffcccf3",
"environmentVersionId": "69447f449bcfd907fc6550c9",
"environmentVersionDescription": "",
"isPublic": true,
"isDownloadable": true,
Expand All @@ -14,8 +14,8 @@
"contextUrl": "https://github.com/datarobot/datarobot-user-models/tree/master/public_dropin_environments/python3_pytorch",
"imageRepository": "env-python-pytorch",
"tags": [
"v11.4.0-6937350034f60512cffcccf3",
"6937350034f60512cffcccf3",
"v11.4.0-69447f449bcfd907fc6550c9",
"69447f449bcfd907fc6550c9",
"v11.4.0-latest"
]
}
6 changes: 3 additions & 3 deletions public_dropin_environments/python3_pytorch/requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
#
# This file is autogenerated by pip-compile with Python 3.10
# This file is autogenerated by pip-compile with Python 3.11
# by the following command:
#
# pip-compile --index-url=https://pypi.org/simple --no-annotate --no-emit-index-url --no-emit-trusted-host --output-file=requirements.txt requirements.in
# pip-compile --cert=None --client-cert=None --index-url=https://pypi.org/simple --no-annotate --no-emit-index-url --no-emit-trusted-host --output-file=requirements.txt --pip-args=None requirements.in
#
annotated-types==0.7.0
argcomplete==3.6.3
Expand All @@ -19,7 +19,7 @@ charset-normalizer==3.4.4
click==8.3.1
cryptography==46.0.3
datarobot==3.10.0
datarobot-drum==1.17.8
datarobot-drum==1.17.10
datarobot-mlops==11.1.0
datarobot-storage==2.2.0
docker==7.1.0
Expand Down
4 changes: 4 additions & 0 deletions public_dropin_environments/python3_sklearn/Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -59,6 +59,10 @@ ENV ADDRESS=0.0.0.0:8080
ENV WITH_ERROR_SERVER=0 \
PYTHONUNBUFFERED=1

# ENV MLOPS_RUNTIME_PARAM_DRUM_CLIENT_REQUEST_TIMEOUT='{"type": "numeric", "payload": 660}'
Copy link

@cursor cursor bot Dec 17, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Bug: Commented-out configuration left in Dockerfile

A commented-out ENV MLOPS_RUNTIME_PARAM_DRUM_CLIENT_REQUEST_TIMEOUT line was added to the Dockerfile. This appears to be leftover testing or debugging code that was not removed before committing. Commented-out configuration in production Dockerfiles can confuse maintainers and suggests incomplete cleanup.

Fix in Cursor Fix in Web


# Gunicorn configuration
ENV MLOPS_RUNTIME_PARAM_DRUM_SERVER_TYPE='{"type": "string", "payload": "gunicorn"}'

COPY ./*.sh ${CODE_DIR}/
WORKDIR ${CODE_DIR}
Expand Down
6 changes: 3 additions & 3 deletions public_dropin_environments/python3_sklearn/env_info.json
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
"description": "This template environment can be used to create artifact-only scikit-learn custom models. This environment contains scikit-learn and only requires your model artifact as a .pkl file and optionally a custom.py file.",
"programmingLanguage": "python",
"label": "",
"environmentVersionId": "6937359c34f60513461b86db",
"environmentVersionId": "69447fb29bcfd90825cad38e",
"environmentVersionDescription": "",
"isPublic": true,
"isDownloadable": true,
Expand All @@ -14,8 +14,8 @@
"contextUrl": "https://github.com/datarobot/datarobot-user-models/tree/master/public_dropin_environments/python3_sklearn",
"imageRepository": "env-python-sklearn",
"tags": [
"v11.4.0-6937359c34f60513461b86db",
"6937359c34f60513461b86db",
"v11.4.0-69447fb29bcfd90825cad38e",
"69447fb29bcfd90825cad38e",
"v11.4.0-latest"
]
}
6 changes: 3 additions & 3 deletions public_dropin_environments/python3_sklearn/requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
#
# This file is autogenerated by pip-compile with Python 3.10
# This file is autogenerated by pip-compile with Python 3.11
# by the following command:
#
# pip-compile --index-url=https://pypi.org/simple --no-annotate --no-emit-index-url --no-emit-trusted-host --output-file=requirements.txt requirements.in
# pip-compile --cert=None --client-cert=None --index-url=https://pypi.org/simple --no-annotate --no-emit-index-url --no-emit-trusted-host --output-file=requirements.txt --pip-args=None requirements.in
#
annotated-types==0.7.0
argcomplete==3.6.3
Expand All @@ -19,7 +19,7 @@ charset-normalizer==3.4.4
click==8.3.1
cryptography==46.0.3
datarobot==3.10.0
datarobot-drum==1.17.8
datarobot-drum==1.17.10
datarobot-mlops==11.1.0
datarobot-storage==2.2.0
docker==7.1.0
Expand Down
3 changes: 3 additions & 0 deletions public_dropin_environments/python3_xgboost/Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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}

Expand Down
6 changes: 3 additions & 3 deletions public_dropin_environments/python3_xgboost/env_info.json
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
"description": "This template environment can be used to create artifact-only xgboost custom models. This environment contains xgboost and only requires your model artifact as a .pkl file and optionally a custom.py file.",
"programmingLanguage": "python",
"label": "",
"environmentVersionId": "693735a534f6051361786f34",
"environmentVersionId": "69447fbb9bcfd9083d7533c2",
"environmentVersionDescription": "",
"isPublic": true,
"isDownloadable": true,
Expand All @@ -14,8 +14,8 @@
"contextUrl": "https://github.com/datarobot/datarobot-user-models/tree/master/public_dropin_environments/python3_xgboost",
"imageRepository": "env-python-xgboost",
"tags": [
"v11.4.0-693735a534f6051361786f34",
"693735a534f6051361786f34",
"v11.4.0-69447fbb9bcfd9083d7533c2",
"69447fbb9bcfd9083d7533c2",
"v11.4.0-latest"
]
}
6 changes: 3 additions & 3 deletions public_dropin_environments/python3_xgboost/requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
#
# This file is autogenerated by pip-compile with Python 3.10
# This file is autogenerated by pip-compile with Python 3.11
# by the following command:
#
# pip-compile --index-url=https://pypi.org/simple --no-annotate --no-emit-index-url --no-emit-trusted-host --output-file=requirements.txt requirements.in
# pip-compile --cert=None --client-cert=None --index-url=https://pypi.org/simple --no-annotate --no-emit-index-url --no-emit-trusted-host --output-file=requirements.txt --pip-args=None requirements.in
#
annotated-types==0.7.0
argcomplete==3.6.3
Expand All @@ -19,7 +19,7 @@ charset-normalizer==3.4.4
click==8.3.1
cryptography==46.0.3
datarobot==3.10.0
datarobot-drum==1.17.8
datarobot-drum==1.17.10
datarobot-mlops==11.1.0
datarobot-storage==2.2.0
docker==7.1.0
Expand Down
3 changes: 3 additions & 0 deletions public_dropin_environments/r_lang/Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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}

Expand Down
Loading