Skip to content

Commit 85fc0fd

Browse files
JCHAVEROTfabnemEPFL
authored andcommitted
chores: fix type check errors in tests
1 parent 31ff215 commit 85fc0fd

1 file changed

Lines changed: 10 additions & 2 deletions

File tree

tests/test_job_queue.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,11 @@ def test_duplicate_file_id_while_in_flight():
4040
manager = JobQueue(devices=["cpu"])
4141
release = threading.Event()
4242

43-
job_id = manager.submit("same", "a.pdf", lambda device: release.wait())
43+
def block(device):
44+
release.wait()
45+
return {}
46+
47+
job_id = manager.submit("same", "a.pdf", block)
4448
_wait_for(manager, job_id, "processing")
4549
with pytest.raises(DuplicateJobError):
4650
manager.submit("same", "b.pdf", lambda device: {})
@@ -54,7 +58,11 @@ def test_queue_full():
5458
manager = JobQueue(devices=["cpu"], max_queue_size=1)
5559
release = threading.Event()
5660

57-
job_id = manager.submit("f1", "a.pdf", lambda device: release.wait())
61+
def block(device):
62+
release.wait()
63+
return {}
64+
65+
job_id = manager.submit("f1", "a.pdf", block)
5866
_wait_for(manager, job_id, "processing")
5967
with pytest.raises(QueueFullError):
6068
manager.submit("f2", "b.pdf", lambda device: {})

0 commit comments

Comments
 (0)