diff --git a/stubs/docutils/docutils/io.pyi b/stubs/docutils/docutils/io.pyi index 5b04fdb71257..182ad5552096 100644 --- a/stubs/docutils/docutils/io.pyi +++ b/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,10 @@ class FileOutput(Output): mode=None, ) -> None: ... def open(self) -> None: ... - def write(self, data): ... + @overload + def write(self, data: str) -> str | bytes: ... + @overload + def write(self, data: bytes) -> bytes: ... def close(self) -> None: ... class BinaryFileOutput(FileOutput): ...