Fix hailo detection (#17819)

This commit is contained in:
Nicolas Mowen 2025-04-19 19:32:14 -06:00 committed by GitHub
parent 484ea10037
commit 43a9641ae8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -28,12 +28,12 @@ class RequestStore:
def put(self, tensor_input: ndarray) -> int:
request_id = self.__get_request_id()
self.input_queue.get((request_id, tensor_input))
self.input_queue.put((request_id, tensor_input))
return request_id
def get(self) -> tuple[int, ndarray] | None:
try:
return self.input_queue.get_nowait()
return self.input_queue.get()
except Exception:
return None