fix parameter mistake

hpcaitech · May 10, 2022 · 81baea8 · 81baea8
1 parent ecee98e
commit 81baea8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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()