Explorar o código

Print max_batch_size in exception

Max Ryabinin %!s(int64=5) %!d(string=hai) anos
pai
achega
36814016f9
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  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()
         task = Task(future1, args)
         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)
         else:
             self.tasks.put(task)