(task_queue, res_queue, num_samples)
| 124 | |
| 125 | |
| 126 | def copy_callback(task_queue, res_queue, num_samples): |
| 127 | msgs = task_queue.get(num_samples=num_samples) |
| 128 | if msgs is None: |
| 129 | return |
| 130 | assert len(msgs) > 0 |
| 131 | res_queue.put(msgs) |
| 132 | return msgs |
| 133 | |
| 134 | |
| 135 | def _test_queue_recv(start_method, worker_params, capacity, send_msgs, recv_msgs, send_one_by_one): |