From cbf11f1b9cf9606fc9b56a48aaefb0bd63baf768 Mon Sep 17 00:00:00 2001 From: Paul Guerrie Date: Tue, 13 Feb 2024 16:07:15 +0000 Subject: [PATCH] Fixed a key error with request IDs --- inference/core/entities/requests/inference.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/inference/core/entities/requests/inference.py b/inference/core/entities/requests/inference.py index e37ab8104b..4bb26742c2 100644 --- a/inference/core/entities/requests/inference.py +++ b/inference/core/entities/requests/inference.py @@ -229,5 +229,5 @@ def request_from_type(model_type, request_dict): request = ObjectDetectionInferenceRequest(**request_dict) else: raise ValueError(f"Uknown task type {model_type}") - request.id = request_dict["id"] + request.id = request_dict.get("id") return request