diff --git a/src/sirocco/core.py b/src/sirocco/core.py index a411104..a856e4d 100644 --- a/src/sirocco/core.py +++ b/src/sirocco/core.py @@ -2,9 +2,9 @@ import logging from dataclasses import dataclass, field -from typing import TYPE_CHECKING, Generic, Self, TypeVar from os.path import expandvars from pathlib import Path +from typing import TYPE_CHECKING, Generic, Self, TypeVar from sirocco.parsing._yaml_data_models import ( ConfigCycleTask, diff --git a/src/sirocco/workgraph.py b/src/sirocco/workgraph.py index 2bf9c73..ce447b3 100644 --- a/src/sirocco/workgraph.py +++ b/src/sirocco/workgraph.py @@ -9,7 +9,6 @@ if TYPE_CHECKING: from aiida_workgraph.socket import TaskSocket # type: ignore[import-untyped] - from wcflow import core @@ -173,7 +172,8 @@ def _add_aiida_task_node(self, task: core.UnrolledTask): def _link_wait_on_to_task(self, task: core.UnrolledTask): # TODO - raise NotImplementedError("") + msg = "" + raise NotImplementedError(msg) label = AiidaWorkGraph.get_aiida_label_from_unrolled_task(task) workgraph_task = self._aiida_task_nodes[label] wait_on_tasks = [] diff --git a/tests/test_wc_workflow.py b/tests/test_wc_workflow.py index 4795744..b7efba8 100644 --- a/tests/test_wc_workflow.py +++ b/tests/test_wc_workflow.py @@ -3,8 +3,8 @@ import pytest from sirocco.core import Workflow -from sirocco.workgraph import AiidaWorkGraph from sirocco.pretty_print import PrettyPrinter +from sirocco.workgraph import AiidaWorkGraph @pytest.fixture @@ -17,7 +17,7 @@ def pprinter(): "tests/files/configs/test_config_large.yml", ] -def test_run_workgraph(config_case, aiida_profile): +def test_run_workgraph(config_case, aiida_profile): # noqa: ARG001 config_path, _ = config_case core_workflow = Workflow.from_yaml(config_path) aiida_workflow = AiidaWorkGraph(core_workflow)