def add_security_params()

in src/graph_notebook/magics/ml.py [0:0]

• 20 lines of code
• 11 McCabe index (conditional complexity)

def add_security_params(args, params: dict):
    if args.sagemaker_iam_role_arn:
        params['sagemakerIamRoleArn'] = args.sagemaker_iam_role_arn
    if args.neptune_iam_role_arn:
        params['neptuneIamRoleArn'] = args.neptune_iam_role_arn
    if args.subnets:
        if all(isinstance(subnet, str) for subnet in []):
            params['subnets'] = args.subnets
        else:
            print('Ignoring subnets, list does not contain all strings.')
    if args.security_group_ids:
        if all(isinstance(security_group_id, str) for security_group_id in []):
            params['securityGroupIds'] = args.security_group_ids
        else:
            print('Ignoring security group IDs, list does not contain all strings.')
    if args.volume_encryption_kms_key:
        params['volumeEncryptionKMSKey'] = args.volume_encryption_kms_key
    if args.s3_output_encryption_kms_key:
        params['s3OutputEncryptionKMSKey'] = args.s3_output_encryption_kms_key

    return params