clear terraform

Author: pirocheto

terraform/main.tf

@@ -4,7 +4,6 @@ locals {
 
 module "model" {
   source               = "./modules/aws-sagemaker-hf-pytorch-inference-model-deployment"
-  aws_region           = "us-east-1"
   model_name           = "${local.env}-multilingual-sentiment-analysis"
   python_version       = "py310"
   ubuntu_version       = "22.04"

terraform/modules/aws-sagemaker-hf-pytorch-inference-model-deployment/main.tf

@@ -1,8 +1,10 @@
 data "aws_caller_identity" "current" {}
 
+data "aws_region" "current" {}
+
 locals {
   account_id     = data.aws_caller_identity.current.account_id
-  ecr_image_name = "763104351884.dkr.ecr.${var.aws_region}.amazonaws.com/huggingface-pytorch-inference"
+  ecr_image_name = "763104351884.dkr.ecr.${data.aws_region.current.name}.amazonaws.com/huggingface-pytorch-inference"
   ecr_image_tag  = "${var.pytorch_version}-transformers${var.transformers_version}-cpu-${var.python_version}-ubuntu${var.ubuntu_version}"
 }
 
@@ -64,7 +66,7 @@ resource "aws_iam_policy" "ecr_policy" {
           "ecr:ListTagsForResource",
           "ecr:DescribeImageScanFindings",
         ],
-        "Resource" : "arn:aws:ecr:${var.aws_region}:*:repository/*"
+        "Resource" : "arn:aws:ecr:${data.aws_region.current.name}:*:repository/*"
       }
     ]
   })
@@ -95,7 +97,7 @@ resource "aws_sagemaker_endpoint_configuration" "model_endpoint_configuration" {
   production_variants {
     initial_variant_weight = 1
     model_name             = aws_sagemaker_model.model.name
-    variant_name           = "Version1"
+    variant_name           = var.endpoint_variant_name
 
     serverless_config {
       memory_size_in_mb = var.memory_size_in_mb

terraform/modules/aws-sagemaker-hf-pytorch-inference-model-deployment/outputs.tf

@@ -2,3 +2,35 @@
 output "sagemaker_endpoint_name" {
   value = aws_sagemaker_endpoint.model_endpoint.name
 }
+
+output "sagemaker_endpoint_arn" {
+  value = aws_sagemaker_endpoint.model_endpoint.arn
+}
+
+output "sagemaker_endpoint_config_name" {
+  value = aws_sagemaker_endpoint_configuration.model_endpoint_configuration.name
+}
+
+output "sagemaker_endpoint_config_arn" {
+  value = aws_sagemaker_endpoint_configuration.model_endpoint_configuration.arn
+}
+
+output "sagemaker_model_name" {
+  value = aws_sagemaker_model.model.name
+}
+
+output "sagemaker_model_arn" {
+  value = aws_sagemaker_model.model.arn
+}
+
+output "sagemaker_model_image" {
+  value = aws_sagemaker_model.model.primary_container.0.image
+}
+
+output "sagemaker_model_data_url" {
+  value = aws_sagemaker_model.model.primary_container.0.model_data_url
+}
+
+output "sagemaker_model_execution_role_arn" {
+  value = aws_sagemaker_model.model.execution_role_arn
+}

terraform/modules/aws-sagemaker-hf-pytorch-inference-model-deployment/variables.tf

@@ -1,14 +1,11 @@
-
-
-variable "aws_region" {
-  description = "AWS region"
-  default     = "us-east-1"
-}
-
 variable "model_name" {
   description = "Name of the model"
 }
 
+variable "endpoint_variant_name" {
+  description = "Name of the endpoint variant"
+}
+
 variable "python_version" {
   description = "Python version"
 }

terraform/provider.tf

@@ -16,5 +16,5 @@ terraform {
 }
 
 provider "aws" {
-  region = var.aws_region
+  region = "us-east-1"
 }