def wait_packaging()

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

• 9 lines of code
• 4 McCabe index (conditional complexity)

    def wait_packaging(self, job):
        """Waits for a Sagemaker Neo packaging job to complete."""
        while True:
            response = sagemaker_client.describe_edge_packaging_job(EdgePackagingJobName=job)
            if response['EdgePackagingJobStatus'] == 'COMPLETED':
                return job, 'Success'
            elif response['EdgePackagingJobStatus'] == 'FAILED':
                return job, 'Failed'
            logger.info('.')
            time.sleep(5)