From 1480bf429555610bc68308342da172ecffc3769c Mon Sep 17 00:00:00 2001 From: Simon Mo Date: Mon, 4 May 2020 20:32:12 -0700 Subject: [PATCH] [Serve] Improve batch size inconsistency error (#8315) --- python/ray/serve/backend_worker.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/python/ray/serve/backend_worker.py b/python/ray/serve/backend_worker.py index 49f2a3d42..524e2b60b 100644 --- a/python/ray/serve/backend_worker.py +++ b/python/ray/serve/backend_worker.py @@ -197,11 +197,12 @@ class RayServeWorker: self.latency_list.append(time.time() - start_timestamp) if (not isinstance(result_list, list)) or (len(result_list) != batch_size): - raise RayServeException("__call__ function " - "doesn't preserve batch-size. " - "Please return a list of result " - "with length equals to the batch " - "size.") + error_message = ("Worker doesn't preserve batch size. The " + "input has length {} but the returned list " + "has length {}. Please return a list of " + "results with length equal to the batch size" + ".".format(batch_size, len(result_list))) + raise RayServeException(error_message) return result_list except Exception as e: wrapped_exception = wrap_to_ray_error(e)