def wait_compilation()

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

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

    def wait_compilation(self, job):
        """Waits for a Sagemaker Neo compilation job to complete."""
        while True:
            response = sagemaker_client.describe_compilation_job(CompilationJobName=job)
            if response['CompilationJobStatus'] == 'COMPLETED':
                return job, 'Success'
            elif response['CompilationJobStatus'] == 'FAILED':
                return job, 'Failed'
            logger.info('.')
            time.sleep(5)