Skip to content

Commit

Permalink
Fix to SageMakerEmbedding class
Browse files Browse the repository at this point in the history
  • Loading branch information
dgallitelli authored Feb 16, 2024
1 parent 669d5bb commit 3eee085
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -47,4 +47,4 @@ def serialize_input(self, request: List[str], model_kwargs: dict) -> bytes:

def deserialize_output(self, response: "StreamingBody") -> List[List[float]]:
response_json = json.loads(response.read().decode("utf-8"))
return response_json["vectors"]
return response_json

0 comments on commit 3eee085

Please sign in to comment.