瀏覽代碼

Print max_batch_size in exception

Max Ryabinin 5 年之前
父節點
當前提交
36814016f9
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      hivemind/runtime/task_pool.py

+ 1 - 1
hivemind/runtime/task_pool.py

@@ -94,7 +94,7 @@ class TaskPool(TaskPoolBase):
         future1, future2 = SharedFuture.make_pair()
         future1, future2 = SharedFuture.make_pair()
         task = Task(future1, args)
         task = Task(future1, args)
         if self.get_task_size(task) > self.max_batch_size:
         if self.get_task_size(task) > self.max_batch_size:
-            exc = ValueError("Task size greater than max_batch_size, it will never be finished")
+            exc = ValueError(f"Task size greater than max_batch_size ({self.max_batch_size}), it will never be finished")
             future2.set_exception(exc)
             future2.set_exception(exc)
         else:
         else:
             self.tasks.put(task)
             self.tasks.put(task)