fix mypy

bridgecrewio · Oct 2, 2023 · b255035 · b255035
1 parent 92e183b
commit b255035
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/checkov/common/bridgecrew/wrapper.py b/checkov/common/bridgecrew/wrapper.py
@@ -199,7 +199,7 @@ def persist_resource_subgraph_maps(
         full_repo_object_key: str,
         timeout: int
 ) -> None:
-    def _upload_resource_subgraph_map(check_type: str, resource_subgraph_map: dict[str, str]):
+    def _upload_resource_subgraph_map(check_type: str, resource_subgraph_map: dict[str, str]) -> None:
         s3_key = os.path.join(graphs_repo_object_key, check_type, "multi-graph/resource_subgraph_maps")
         try:
             _put_json_object(s3_client, resource_subgraph_map, bucket, s3_key)

diff --git a/checkov/common/runners/runner_registry.py b/checkov/common/runners/runner_registry.py
@@ -775,4 +775,4 @@ def extract_graphs_from_runner(runner: _BaseRunner) -> list[tuple[LibraryGraph,
     @staticmethod
     def extract_resource_subgraph_map_from_runner(runner: _BaseRunner) -> Optional[dict[str, str]]:
         # exist only for terraform
-        return getattr(runner, 'resource_subgraph_map', None)   # type:ignore[no-any-return]
+        return getattr(runner, 'resource_subgraph_map', None)
diff --git a/checkov/terraform/graph_manager.py b/checkov/terraform/graph_manager.py
@@ -129,7 +129,9 @@ def build_multi_graph_from_definitions(self, definitions: dict[TFDefinitionKey,
         return graphs
 
     @staticmethod
-    def update_resource_subgraph_map(local_graph: TerraformLocalGraph, subgraph_path: str, resource_subgraph_map: dict[str, str], source_dir: str):
+    def update_resource_subgraph_map(
+            local_graph: TerraformLocalGraph, subgraph_path: str, resource_subgraph_map: dict[str, str], source_dir: str
+    ) -> None:
         for v in local_graph.vertices:
             resource_id = f"/{os.path.relpath(v.path, source_dir)}:{v.id}"
             resource_subgraph_map[resource_id] = subgraph_path