diff --git a/engines/pyfunc-ensembler-service/app.Dockerfile b/engines/pyfunc-ensembler-service/app.Dockerfile index 61e4867c7..b0d0220ee 100644 --- a/engines/pyfunc-ensembler-service/app.Dockerfile +++ b/engines/pyfunc-ensembler-service/app.Dockerfile @@ -52,4 +52,4 @@ RUN if [ "${MLFLOW_ARTIFACT_STORAGE_TYPE}" = "gcs" ]; then \ ENV FOLDER_NAME=$FOLDER_NAME SHELL ["/bin/bash", "-c"] ENTRYPOINT . activate ${CONDA_ENV_NAME} && \ - python -m pyfunc_ensembler_runner --mlflow_ensembler_dir ./${FOLDER_NAME} -l INFO + python -m pyfunc_ensembler_runner --mlflow_ensembler_dir ./${FOLDER_NAME} diff --git a/engines/pyfunc-ensembler-service/pyfunc_ensembler_runner/__main__.py b/engines/pyfunc-ensembler-service/pyfunc_ensembler_runner/__main__.py index caddb7ee3..fbeb737bb 100644 --- a/engines/pyfunc-ensembler-service/pyfunc_ensembler_runner/__main__.py +++ b/engines/pyfunc-ensembler-service/pyfunc_ensembler_runner/__main__.py @@ -1,6 +1,7 @@ import argparse import logging import traceback +import os import tornado.ioloop @@ -22,20 +23,12 @@ help="Dry run pyfunc ensembler by loading the specified ensembler " "in --mlflow_ensembler_dir without starting webserver", ) -parser.add_argument( - "-l", - "--log-level", - dest="log_level", - choices=["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"], - help="Set the logging level", - default=logging.DEBUG, -) args, _ = parser.parse_known_args() - +log_level = os.getenv("LOG_LEVEL", logging.WARNING) if __name__ == "__main__": - logging.basicConfig(level=args.log_level) + logging.basicConfig(level=log_level) logging.info( "Called with arguments:\n%s\n", "\n".join([f"{k}: {v}" for k, v in vars(args).items()]), diff --git a/engines/pyfunc-ensembler-service/tests/test_ensembler_runner.py b/engines/pyfunc-ensembler-service/tests/test_ensembler_runner.py index fc5ebd781..fdf246217 100644 --- a/engines/pyfunc-ensembler-service/tests/test_ensembler_runner.py +++ b/engines/pyfunc-ensembler-service/tests/test_ensembler_runner.py @@ -27,7 +27,7 @@ "simple_ensembler_uri", dummy_short_request, {"Key": "Value"}, - {"headers": {"Key": "Value"}, "enricher_response" : None}, + {"headers": {"Key": "Value"}, "enricher_response": None}, ), ], )