def package_model()

in sample-apps/custom-model/code/model/model.py [0:0]

• 17 lines of code
• 1 McCabe index (conditional complexity)

    def package_model(self, model_name, compilation_job):
        """Packages a model with Amazon Sagemaker Neo."""
        PACKAGING_JOB=compilation_job+"-packaging"
        MODEL_VERSION = "1.0"
        PACKAGED_MODEL_FOLDER_URI = 's3://{}/models-packaged'.format(self.bucket_name)
        MODEL_PACKAGE = '{}-{}.tar.gz'.format(model_name, MODEL_VERSION)
        response = sagemaker_client.create_edge_packaging_job(
            RoleArn=self.service_role_arn,
            OutputConfig={
                'S3OutputLocation': PACKAGED_MODEL_FOLDER_URI,
            },
            ModelName=model_name,
            ModelVersion=MODEL_VERSION,
            EdgePackagingJobName=PACKAGING_JOB,
            CompilationJobName=compilation_job,
        )
        logger.info(response)
        return PACKAGING_JOB