Skip to content

Commit 8117a45

Browse files
committed
fix drum
1 parent b08a6dd commit 8117a45

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

custom_model_runner/datarobot_drum/drum/entry_point.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,14 @@
11
from datarobot_drum.drum.gunicorn.run_gunicorn import main_gunicorn
22
from datarobot_drum.drum.main import main
33
from datarobot_drum import RuntimeParameters
4+
from datarobot_drum.drum.utils.setup import setup_options
45

56

67
def run_drum_server():
8+
options = setup_options()
79
if (
8-
RuntimeParameters.has("DRUM_SERVER_TYPE")
10+
options.subparser_name == "server"
11+
and RuntimeParameters.has("DRUM_SERVER_TYPE")
912
and str(RuntimeParameters.get("DRUM_SERVER_TYPE")).lower() == "gunicorn"
1013
):
1114
main_gunicorn()

0 commit comments

Comments
 (0)