Fix assignement error

adap · Sep 28, 2024 · decf065 · decf065
1 parent 5ad958d
commit decf065
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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]