def _adapt_to_ts_format()

in src/sagemaker_pytorch_serving_container/torchserve.py [0:0]

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

def _adapt_to_ts_format(handler_service):
    if not os.path.exists(DEFAULT_TS_MODEL_DIRECTORY):
        os.makedirs(DEFAULT_TS_MODEL_DIRECTORY)

    model_archiver_cmd = [
        "torch-model-archiver",
        "--model-name",
        DEFAULT_TS_MODEL_NAME,
        "--handler",
        handler_service,
        "--export-path",
        DEFAULT_TS_MODEL_DIRECTORY,
        "--version",
        "1",
        "--extra-files",
        os.path.join(environment.model_dir)
    ]

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

    _set_python_path()