소스 검색

priority pool

Your Name 2 년 전
부모
커밋
084d565845
1개의 변경된 파일2개의 추가작업 그리고 1개의 파일을 삭제
  1. 2 1
      tests/test_priority_pool.py

+ 2 - 1
tests/test_priority_pool.py

@@ -13,7 +13,8 @@ def test_priority_pools():
     outputs_queue = mp.SimpleQueue()
     results_valid = mp.Event()
 
-    def dummy_pool_func(x):
+    def dummy_pool_func(args, kwargs):
+        (x,) = args  # TODO modify the PriorityPool code such that dummy_pool_func can accept x directly
         time.sleep(0.1)
         y = x**2
         outputs_queue.put((x, y))