fix: some small type hints fixes

haidaraM · Feb 12, 2022 · 89a526d · 89a526d
1 parent 0e16ceb
commit 89a526d
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/ansibleplaybookgrapher/cli.py b/ansibleplaybookgrapher/cli.py
@@ -160,7 +160,7 @@ def _add_my_options(self):
             "--open-protocol-custom-formats",
             dest="open_protocol_custom_formats",
             default=None,
-            help="""The custom formats to use as URLs for the nodes in the graph. Required if 
+            help="""The custom formats to use as URLs for the nodes in the graph. Required if
                                  --open-protocol-handler is set to custom.
                                  You should provide a JSON formatted string like: {"file": "", "folder": ""}.
                                  Example: If you want to open folders (roles) inside the browser and files (tasks) in

diff --git a/ansibleplaybookgrapher/graph.py b/ansibleplaybookgrapher/graph.py
@@ -112,7 +112,7 @@ def links_structure(self) -> Dict[Node, List[Node]]:
         list of the linked nodes
         :return:
         """
-        links = defaultdict(list)
+        links: Dict[Node, List[Node]] = defaultdict(list)
         self._get_all_links(links)
         return links
 
@@ -364,6 +364,6 @@ def get_all_tasks_nodes(composite: CompositeNode) -> List[TaskNode]:
     :param composite:
     :return:
     """
-    tasks = []
+    tasks: List[TaskNode] = []
     _get_all_tasks_nodes(composite, tasks)
     return tasks
diff --git a/ansibleplaybookgrapher/parser.py b/ansibleplaybookgrapher/parser.py
@@ -282,7 +282,7 @@ def _include_tasks_in_blocks(
         parent_nodes: List[CompositeNode],
         block: Union[Block, TaskInclude],
         node_type: str,
-        play_vars: Dict = None,
+        play_vars: Dict,
     ):
         """
         Recursively read all the tasks of the block and add it to the graph