Add missing type annotation

praw-dev · Nov 28, 2022 · bb8c7a7 · bb8c7a7
1 parent 239ce85
commit bb8c7a7
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/asyncpraw/models/reddit/mixins/replyable.py b/asyncpraw/models/reddit/mixins/replyable.py
@@ -1,5 +1,5 @@
 """Provide the ReplyableMixin class."""
-from typing import TYPE_CHECKING, Optional
+from typing import TYPE_CHECKING, Optional, Union
 
 from ....const import API_PATH
 
@@ -10,13 +10,15 @@
 class ReplyableMixin:
     """Interface for :class:`.RedditBase` classes that can be replied to."""
 
-    async def reply(self, body: str) -> Optional["asyncpraw.models.Comment"]:
+    async def reply(
+        self, body: str
+    ) -> Optional[Union["asyncpraw.models.Comment", "asyncpraw.models.Message"]]:
         """Reply to the object.
 
         :param body: The Markdown formatted content for a comment.
 
-        :returns: A :class:`.Comment` object for the newly created comment or ``None``
-            if Reddit doesn't provide one.
+        :returns: A :class:`.Comment` or :class:`.Message` object for the newly created
+            comment or message or ``None`` if Reddit doesn't provide one.
 
         :raises: ``asyncprawcore.exceptions.Forbidden`` when attempting to reply to some
             items, such as locked submissions/comments or non-replyable messages.