Skip to content

Commit 12f3c3d

Browse files
committed
fix test
1 parent e9e3c1b commit 12f3c3d

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

.gitignore

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,6 @@
11
.DS_Store
22

33
.terraform/
4-
.terraform.lock.hcl
4+
.terraform.lock.hcl
5+
6+
.mypy_cache/

test_model/test_endpoint.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,12 @@
11
import json
2+
import os
23
import time
34

45
import boto3
56

67
AWS_REGION = "us-east-1"
7-
ENDPOINT_NAME = "prod-multilingual-sentiment-analysis-endpoint"
8+
ENVIRONMENT = os.environ.get("ENV", "dev")
9+
ENDPOINT_NAME = f"{ENVIRONMENT}-multilingual-sentiment-analysis-endpoint"
810

911
sagemaker_runtime = boto3.client("sagemaker-runtime", region_name=AWS_REGION)
1012

0 commit comments

Comments
 (0)