def _adapt_to_mms_format()

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

• 12 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)