diff --git a/src/py/flwr/server/superlink/state/state_test.py b/src/py/flwr/server/superlink/state/state_test.py index 85cda1a5af9c..606aeb926ccf 100644 --- a/src/py/flwr/server/superlink/state/state_test.py +++ b/src/py/flwr/server/superlink/state/state_test.py @@ -315,8 +315,8 @@ def test_task_res_store_and_retrieve_by_task_ins_id(self) -> None: # Execute task_res_uuid = state.store_task_res(task_res) - if task_ins_id is not None: - task_res_list = state.get_task_res(task_ids={task_ins_id}, limit=None) + assert task_ins_id + task_res_list = state.get_task_res(task_ids={task_ins_id}, limit=None) # Assert retrieved_task_res = task_res_list[0]