File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -116,6 +116,7 @@ def test_dependency_steps(self):
116116 fs1 = Future ()
117117 fs2 = Future ()
118118 q = Queue ()
119+ qr = Queue ()
119120 q .put (
120121 {
121122 "fn" : add_function ,
@@ -150,6 +151,7 @@ def test_dependency_steps(self):
150151 target = _execute_tasks_with_dependencies ,
151152 kwargs = {
152153 "future_queue" : q ,
154+ "return_queue" : qr ,
153155 "executor_queue" : executor ._future_queue ,
154156 "executor" : executor ,
155157 "refresh_rate" : 0.01 ,
@@ -168,6 +170,7 @@ def test_dependency_steps_error(self):
168170 fs1 = Future ()
169171 fs2 = Future ()
170172 q = Queue ()
173+ qr = Queue ()
171174 q .put (
172175 {
173176 "fn" : raise_error ,
@@ -202,6 +205,7 @@ def test_dependency_steps_error(self):
202205 target = _execute_tasks_with_dependencies ,
203206 kwargs = {
204207 "future_queue" : q ,
208+
205209 "executor_queue" : executor ._future_queue ,
206210 "executor" : executor ,
207211 "refresh_rate" : 0.01 ,
@@ -222,6 +226,7 @@ def test_dependency_steps_error_before(self):
222226 fs1 .set_exception (RuntimeError ())
223227 fs2 = Future ()
224228 q = Queue ()
229+ qr = Queue ()
225230 q .put (
226231 {
227232 "fn" : add_function ,
@@ -256,6 +261,7 @@ def test_dependency_steps_error_before(self):
256261 target = _execute_tasks_with_dependencies ,
257262 kwargs = {
258263 "future_queue" : q ,
264+ "return_queue" : qr ,
259265 "executor_queue" : executor ._future_queue ,
260266 "executor" : executor ,
261267 "refresh_rate" : 0.01 ,
You can’t perform that action at this time.
0 commit comments