Improve type for yaml composer (#5144)

ssbarnea · web-flow · commit 2d3bf41b1849 · 2021-04-08T07:29:11.000-07:00
diff --git a/stubs/PyYAML/yaml/composer.pyi b/stubs/PyYAML/yaml/composer.pyi
@@ -1,17 +1,18 @@
-from typing import Any
+from typing import Any, Dict, Optional, Union
 
 from yaml.error import MarkedYAMLError
+from yaml.nodes import MappingNode, Node, ScalarNode, SequenceNode
 
 class ComposerError(MarkedYAMLError): ...
 
 class Composer:
-    anchors: Any
+    anchors: Dict[Any, Node]
     def __init__(self) -> None: ...
-    def check_node(self): ...
-    def get_node(self): ...
-    def get_single_node(self): ...
-    def compose_document(self): ...
-    def compose_node(self, parent, index): ...
-    def compose_scalar_node(self, anchor): ...
-    def compose_sequence_node(self, anchor): ...
-    def compose_mapping_node(self, anchor): ...
+    def check_node(self) -> bool: ...
+    def get_node(self) -> Optional[Node]: ...
+    def get_single_node(self) -> Optional[Node]: ...
+    def compose_document(self) -> Optional[Node]: ...
+    def compose_node(self, parent: Optional[Node], index: int) -> Optional[Node]: ...
+    def compose_scalar_node(self, anchor: Dict[Any, Node]) -> ScalarNode: ...
+    def compose_sequence_node(self, anchor: Dict[Any, Node]) -> SequenceNode: ...
+    def compose_mapping_node(self, anchor: Dict[Any, Node]) -> MappingNode: ...
