From 2d3bf41b1849ee0816da87b07a5224ff47c68825 Mon Sep 17 00:00:00 2001
From: Sorin Sbarnea <sorin.sbarnea@gmail.com>
Date: Thu, 8 Apr 2021 15:29:11 +0100
Subject: [PATCH] Improve type for yaml composer (#5144)

---
 stubs/PyYAML/yaml/composer.pyi | 21 +++++++++++----------
 1 file changed, 11 insertions(+), 10 deletions(-)

diff --git a/stubs/PyYAML/yaml/composer.pyi b/stubs/PyYAML/yaml/composer.pyi
index c367f50b5..a2904ff98 100644
--- 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: ...