def _adapt_to_mms_format()

in src/sagemaker_xgboost_container/mms_patch/model_server.py [0:0]

• 18 lines of code
• 2 McCabe index (conditional complexity)

def _adapt_to_mms_format(handler_service):
    """Archive initial model using MMS handler
    :param handler_service:
    :return:
    """
    if not os.path.exists(DEFAULT_MMS_MODEL_DIRECTORY):
        os.makedirs(DEFAULT_MMS_MODEL_DIRECTORY)

    model_archiver_cmd = [
        "model-archiver",
        "--model-name",
        DEFAULT_MMS_MODEL_NAME,
        "--handler",
        handler_service,
        "--model-path",
        environment.model_dir,
        "--export-path",
        DEFAULT_MMS_MODEL_DIRECTORY,
        "--archive-format",
        "no-archive",
    ]

    logger.info(model_archiver_cmd)
    subprocess.check_call(model_archiver_cmd)