-
Notifications
You must be signed in to change notification settings - Fork 0
/
stepfunctions.tf
40 lines (34 loc) · 1.57 KB
/
stepfunctions.tf
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
locals {
lambda_arn_prefix = "arn:aws:lambda:${var.AWS_REGION}:${data.aws_caller_identity.current.account_id}:function"
}
# Template file for training step function definition
data "template_file" "training_template" {
template = file("${path.cwd}/config/training-stepfunction.json")
vars = {
account_id = data.aws_caller_identity.current.account_id
region = var.AWS_REGION
service_name = "${var.SERVICE}-${var.BUILD_STAGE}"
sagemaker_role_arn = aws_iam_role.sagemaker_role.arn
}
}
resource "aws_sfn_state_machine" "training_step_function" {
name = "${var.SERVICE}-${var.BUILD_STAGE}-training-step-function"
role_arn = aws_iam_role.step_function_role.arn
definition = data.template_file.training_template.rendered
}
# Template file for inference step function definition
data "template_file" "inference_template" {
template = file("${path.cwd}/config/inference-stepfunction.json")
vars = {
account_id = data.aws_caller_identity.current.account_id
region = var.AWS_REGION
service_name = "${var.SERVICE}-${var.BUILD_STAGE}"
sagemaker_role_arn = aws_iam_role.sagemaker_role.arn
check_endpoint_status_lambda = "${local.lambda_arn_prefix}:${aws_ssm_parameter.update_endpoint_status_lambda_name.value}"
}
}
resource "aws_sfn_state_machine" "inference_step_function" {
name = "${var.SERVICE}-${var.BUILD_STAGE}-inference-step-function"
role_arn = aws_iam_role.step_function_role.arn
definition = data.template_file.inference_template.rendered
}