Enabling variables to control job batch limits

main.tf

@@ -97,6 +97,10 @@ module "metaflow-computation" {
   launch_template_http_endpoint               = var.launch_template_http_endpoint
   launch_template_http_tokens                 = var.launch_template_http_tokens
   launch_template_http_put_response_hop_limit = var.launch_template_http_put_response_hop_limit
+  job_state_time_limit_action                 = var.job_state_time_limit_action
+  job_state_time_limit_timeout                = var.job_state_time_limit_timeout
+  job_state_time_limit_reason                 = var.job_state_time_limit_reason
+
 
   standard_tags = var.tags
 }

modules/computation/variables.tf

@@ -102,3 +102,18 @@ variable "launch_template_image_id" {
   nullable    = true
   default     = null
 }
+
+variable "job_state_time_limit_action" {
+  type        = string
+  description = "The action to take when the job times out"
+}
+
+variable "job_state_time_limit_timeout" {
+  type        = number
+  description = "The time limit in seconds for the job to run before the action is taken"
+}
+
+variable "job_state_time_limit_reason" {
+  type        = number
+  description = "The reason for the job state time limit action"
+}

variables.tf

@@ -199,3 +199,18 @@ variable "enable_key_rotation" {
   description = "Enable key rotation for KMS keys"
   default     = false
 }
+
+variable "job_state_time_limit_action" {
+  type        = string
+  description = "The action to take when the job times out"
+}
+
+variable "job_state_time_limit_timeout" {
+  type        = number
+  description = "The time limit in seconds for the job to run before the action is taken"
+}
+
+variable "job_state_time_limit_reason" {
+  type        = number
+  description = "The reason for the job state time limit action"
+}