def poll_training_status()

in infra/src/lambda_poller/frauddetector_poller.py [0:0]

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

    def poll_training_status(self, job_parameters, client=None):
        result_is_job_complete = False
        client = client or boto3.client('frauddetector')

        training_job_name = job_parameters['training_job_name']
        training_job_version = job_parameters['training_job_version']

        response = client.get_model_version(modelId=training_job_name,
                                            modelType='ONLINE_FRAUD_INSIGHTS',
                                            modelVersionNumber=training_job_version)
        training_job_status = response["status"]
        print(response)

        # Flag CodePipeline status
        if training_job_status == 'TRAINING_COMPLETE':
            result_is_job_complete = True

        elif training_job_status == 'ERROR':
            raise Exception(f'Training job: {training_job_name} failed {response}')

        return result_is_job_complete