You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I have a trained sklearn model deployed as a sagemaker endpoint that I want to invoke from the aurora postgres extenstion using aws_sagemaker.invoke_endpoint. Invoking works correctly as long as each batch has more than one sample but if the batch has one sample it returns a 500 with the error ValueError: Expected 2D array, got 1D array instead.
For example if there are 100 samples and I call the postgres extention with a max_batch_size of 10 like below it will score 10 batches without issue. If there were 101 samples it would score 10 batches successfully and then blow up on the 11th.
I have a trained sklearn model deployed as a sagemaker endpoint that I want to invoke from the aurora postgres extenstion using
aws_sagemaker.invoke_endpoint
. Invoking works correctly as long as each batch has more than one sample but if the batch has one sample it returns a 500 with the errorValueError: Expected 2D array, got 1D array instead
.For example if there are 100 samples and I call the postgres extention with a
max_batch_size
of 10 like below it will score 10 batches without issue. If there were 101 samples it would score 10 batches successfully and then blow up on the 11th.I saw issue #15 had a similar problem invoking from the api but I'm not sure how I can pass in an extra line when invoking the endpoint.
The text was updated successfully, but these errors were encountered: