Skip to content

Commit

Permalink
Lint fix
Browse files Browse the repository at this point in the history
Signed-off-by: <>
  • Loading branch information
jitu5 committed Apr 25, 2024
1 parent 15db91b commit 2e15a47
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 5 deletions.
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
"""`kedro_viz.data_access.repositories.modular_pipelines`
defines repository to centralise access to modular pipelines data."""

from typing import Dict, Optional, Set, Union

from kedro_viz.constants import ROOT_MODULAR_PIPELINE_ID
Expand Down Expand Up @@ -224,7 +225,7 @@ def extract_from_node(self, node: GraphNode) -> Optional[str]:
ModularPipelineChild(id=node.id, type=GraphNodeType(node.type)),
)
return modular_pipeline_id

def inherit_tags_recursive(self, modular_pipeline_id: str, tags: Set[str]):
"""Recursively collects a set of tags from a modular pipeline to all of its
child modular pipelines.
Expand All @@ -238,7 +239,7 @@ def inherit_tags_recursive(self, modular_pipeline_id: str, tags: Set[str]):
modular_pipeline.tags.update(tags)
for child in modular_pipeline.children:
if child.type == GraphNodeType.MODULAR_PIPELINE:
self.inherit_tags_recursive(child.id, modular_pipeline.tags)
self.inherit_tags_recursive(child.id, modular_pipeline.tags)

def has_modular_pipeline(self, modular_pipeline_id: str) -> bool:
"""Return whether this modular pipeline repository has a given modular pipeline ID.
Expand Down
5 changes: 2 additions & 3 deletions package/kedro_viz/services/modular_pipelines.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
"""`kedro_viz.services.modular_pipelines` defines modular pipelines-related business logic.
The service layer consist of pure functions operating on domain models.
"""

from typing import Dict

from kedro_viz.constants import ROOT_MODULAR_PIPELINE_ID
Expand Down Expand Up @@ -96,7 +97,5 @@ def expand_tree(
expanded_tree[parent_id].external_outputs.update(
modular_pipeline_node.external_outputs
)
expanded_tree[parent_id].tags.update(
modular_pipeline_node.tags
)
expanded_tree[parent_id].tags.update(modular_pipeline_node.tags)
return expanded_tree

0 comments on commit 2e15a47

Please sign in to comment.