From 814ee42ce3f2d43a6ade28a5a2f7ac7f83ad420d Mon Sep 17 00:00:00 2001 From: Haotian Zhang Date: Fri, 16 Feb 2024 10:21:15 -0500 Subject: [PATCH] cr --- .../llama_index/embeddings/sagemaker_endpoint/utils.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/llama-index-integrations/embeddings/llama-index-embeddings-sagemaker-endpoint/llama_index/embeddings/sagemaker_endpoint/utils.py b/llama-index-integrations/embeddings/llama-index-embeddings-sagemaker-endpoint/llama_index/embeddings/sagemaker_endpoint/utils.py index dd0346056ae31..4f7602ebde74e 100644 --- a/llama-index-integrations/embeddings/llama-index-embeddings-sagemaker-endpoint/llama_index/embeddings/sagemaker_endpoint/utils.py +++ b/llama-index-integrations/embeddings/llama-index-embeddings-sagemaker-endpoint/llama_index/embeddings/sagemaker_endpoint/utils.py @@ -46,5 +46,4 @@ def serialize_input(self, request: List[str], model_kwargs: dict) -> bytes: return request_str.encode("utf-8") def deserialize_output(self, response: "StreamingBody") -> List[List[float]]: - response_json = json.loads(response.read().decode("utf-8")) - return response_json + return json.loads(response.read().decode("utf-8"))