diff --git a/pyproject.toml b/pyproject.toml index 08356a0e..f93f48bb 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -33,7 +33,7 @@ dependencies = [ "aiida-core>=2.3", "cloudpickle", "aiida-shell~=0.8", - "aiida-pythonjob==0.1.4", + "aiida-pythonjob>=0.1.6", "fastapi", "uvicorn", "pydantic_settings", diff --git a/tests/test_action.py b/tests/test_action.py index ba403877..d8f98c8b 100644 --- a/tests/test_action.py +++ b/tests/test_action.py @@ -22,14 +22,14 @@ def test_pause_play_task(wg_calcjob): wg.pause_tasks(["add1"]) wg.submit() # wait for the workgraph to launch add1 - wg.wait(tasks={"add1": ["CREATED"]}, timeout=40, interval=5) + wg.wait(tasks={"add1": ["CREATED"]}, timeout=60, interval=5) assert wg.tasks.add1.node.process_state.value.upper() == "CREATED" assert wg.tasks.add1.node.process_status == "Paused through WorkGraph" # pause add2 after submit wg.pause_tasks(["add2"]) wg.play_tasks(["add1"]) # wait for the workgraph to launch add2 - wg.wait(tasks={"add2": ["CREATED"]}, timeout=40, interval=5) + wg.wait(tasks={"add2": ["CREATED"]}, timeout=60, interval=5) assert wg.tasks.add2.node.process_state.value.upper() == "CREATED" assert wg.tasks.add2.node.process_status == "Paused through WorkGraph" # I disabled the following lines because the test is not stable