소스 검색

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()
         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)