Type FileOutput.write

Author: adamtheturtle

stubs/docutils/docutils/io.pyi

@@ -8,7 +8,7 @@ from _typeshed import (
     Unused,
 )
 from re import Pattern
-from typing import IO, Any, ClassVar, Final, Generic, Literal, TypeVar
+from typing import IO, Any, ClassVar, Final, Generic, Literal, TypeVar, overload
 
 from docutils import TransformSpec, nodes
 
@@ -104,7 +104,12 @@ class FileOutput(Output):
         mode=None,
     ) -> None: ...
     def open(self) -> None: ...
-    def write(self, data): ...
+    @overload
+    def write(self, data: str) -> str: ...
+
+    @overload
+    def write(self, data: bytes) -> bytes: ...
+    def write(self, data: str | bytes) -> str | bytes: ...
     def close(self) -> None: ...
 
 class BinaryFileOutput(FileOutput): ...