From 81baea8065c7f4f51223d9bb0116c8d6fd31c867 Mon Sep 17 00:00:00 2001 From: maruyama Date: Tue, 10 May 2022 16:06:25 +0800 Subject: [PATCH] fix parameter mistake --- energon/server/batch_manager.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/energon/server/batch_manager.py b/energon/server/batch_manager.py index 3403360..37b8a04 100644 --- a/energon/server/batch_manager.py +++ b/energon/server/batch_manager.py @@ -242,7 +242,7 @@ def processing_batch(self): else: input_ids = input_text output = self.engine.run(input_ids) - self.pool.submit(self.publish_result, output, target_batch, start_time) + self.pool.submit(self.publish_result, output, target_batch) # self.publish_result(output, target_batch, start_time) # pub_thread = threading.Thread(target=self.publish_result, args=(output, target_batch, start_time)) # pub_thread.start()