From 3fbfe8e6cc392ce9a22d0b059a8b768e10e6c44e Mon Sep 17 00:00:00 2001 From: Stephen Roller <roller@fb.com> Date: Fri, 22 Jul 2022 15:01:41 +0000 Subject: [PATCH] [api] Fix worker queue logic --- metaseq_cli/interactive_hosted.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/metaseq_cli/interactive_hosted.py b/metaseq_cli/interactive_hosted.py index 2b9e37a..2365936 100644 --- a/metaseq_cli/interactive_hosted.py +++ b/metaseq_cli/interactive_hosted.py @@ -105,8 +105,8 @@ def batching_loop(timeout=100, max_tokens=MAX_BATCH_TOKENS): # batch is empty or under budget batch.append(item) except queue.Empty: + target_queue = None if batch: - target_queue = None request_object = { "inputs": [], "min_tokens": [], -- GitLab