Skip to content

Commit

Permalink
fix some fmt
Browse files Browse the repository at this point in the history
  • Loading branch information
agoscinski committed Nov 26, 2024
1 parent 8d4bba3 commit af95c0a
Show file tree
Hide file tree
Showing 3 changed files with 5 additions and 5 deletions.
2 changes: 1 addition & 1 deletion src/sirocco/core.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down
4 changes: 2 additions & 2 deletions src/sirocco/workgraph.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,6 @@

if TYPE_CHECKING:
from aiida_workgraph.socket import TaskSocket # type: ignore[import-untyped]

from wcflow import core


Expand Down Expand Up @@ -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 = []
Expand Down
4 changes: 2 additions & 2 deletions tests/test_wc_workflow.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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)
Expand Down

0 comments on commit af95c0a

Please sign in to comment.