From b3cbedbaa00e127aefd2f8d7b9b067f33bbc9e26 Mon Sep 17 00:00:00 2001 From: crusaderky Date: Tue, 5 Apr 2022 17:34:25 +0100 Subject: [PATCH] Polish --- distributed/tests/test_worker_state_machine.py | 2 +- distributed/worker_state_machine.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/distributed/tests/test_worker_state_machine.py b/distributed/tests/test_worker_state_machine.py index 8cb6c73a7e5..87828a17cf7 100644 --- a/distributed/tests/test_worker_state_machine.py +++ b/distributed/tests/test_worker_state_machine.py @@ -131,5 +131,5 @@ def test_merge_recs_instructions(): {x: "memory"}, [], ) - with pytest.raises(AssertionError): + with pytest.raises(ValueError): merge_recs_instructions(({x: "memory"}, []), ({x: "released"}, [])) diff --git a/distributed/worker_state_machine.py b/distributed/worker_state_machine.py index 35b4a2130cc..d3ff40ebaf4 100644 --- a/distributed/worker_state_machine.py +++ b/distributed/worker_state_machine.py @@ -423,8 +423,8 @@ def merge_recs_instructions(*args: RecsInstrs) -> RecsInstrs: for recs_i, instr_i in args: for k, v in recs_i.items(): if k in recs and recs[k] != v: - raise AssertionError( # pragma: nocover - f"Mismatched reccommendations for {k}: {recs[k]} vs. {v}" + raise ValueError( + f"Mismatched recommendations for {k}: {recs[k]} vs. {v}" ) recs[k] = v instr += instr_i