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

Fix tests on sagemaker.py
pull/465/head
Pavel 1 year ago committed by GitHub
commit 68b8d7d7f2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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'
)

Loading…
Cancel
Save