def get_dev_config()

in model/run_pipeline.py [0:0]

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

def get_dev_config(model_name, job_id, role, image_uri, kms_key_id, sagemaker_project_id):
    return {
        "Parameters": {
            "ImageRepoUri": image_uri,
            "ModelName": model_name,
            "TrainJobId": job_id,
            "DeployRoleArn": role,
            "ModelVariant": "dev",
            "KmsKeyId": kms_key_id,
        },
        "Tags": {
            "mlops:model-name": model_name,
            "mlops:stage": "dev",
            "SageMakerProjectId": sagemaker_project_id,
        },
    }