Merge pull request #464 from arc53/bug/fix-tests

Fix tests on sagemaker.py
This commit is contained in:
Pavel
2023-10-06 17:02:54 +03:00
committed by GitHub

View File

@@ -64,9 +64,9 @@ class SagemakerAPILLM(BaseLLM):
import boto3
runtime = boto3.client(
'runtime.sagemaker',
aws_access_key_id=settings.SAGEMAKER_ACCESS_KEY,
aws_secret_access_key=settings.SAGEMAKER_SECRET_KEY,
region_name=settings.SAGEMAKER_REGION
aws_access_key_id='xxx',
aws_secret_access_key='xxx',
region_name='us-west-2'
)