From 075e704beae622cb2727ca23aeff5b5fbf91366d Mon Sep 17 00:00:00 2001 From: Barak Fatal Date: Tue, 26 Sep 2023 11:02:24 +0300 Subject: [PATCH] Mypy --- checkov/common/images/graph/image_referencer_provider.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/checkov/common/images/graph/image_referencer_provider.py b/checkov/common/images/graph/image_referencer_provider.py index 5f675bb2556..25d5cb99841 100644 --- a/checkov/common/images/graph/image_referencer_provider.py +++ b/checkov/common/images/graph/image_referencer_provider.py @@ -2,6 +2,7 @@ import itertools import os +import typing from abc import abstractmethod from typing import TYPE_CHECKING, Callable, Any, Mapping, Generator @@ -41,6 +42,8 @@ def extract_nodes(self) -> LibraryGraph | None: return self.extract_nodes_networkx() def extract_nodes_networkx(self) -> networkx.Graph: + if typing.TYPE_CHECKING: + self.graph_connector = typing.cast(networkx.Graph, self.graph_connector) resource_nodes = [ node for node, resource_type in self.graph_connector.nodes(data=CustomAttributes.RESOURCE_TYPE) @@ -56,9 +59,11 @@ def extract_nodes_rustworkx(self) -> rustworkx.PyDiGraph[Any, Any]: if self.resource_type_pred(node, list(self.supported_resource_types)) ] - return self.graph_connector.subgraph(resource_nodes) # type: ignore + return self.graph_connector.subgraph(resource_nodes) def extract_nodes_igraph(self) -> igraph.Graph: + if typing.TYPE_CHECKING: + self.graph_connector = typing.cast(igraph.Graph, self.graph_connector) resource_nodes = [ node for node, resource_type in itertools.zip_longest(self.graph_connector.vs['name'],