Add type annotations

2 files changed, 2 insertions, 2 deletions

diff --git a/nbxmpp/protocol.py b/nbxmpp/protocol.py
index 4d73135..15a9bbd 100644
--- a/nbxmpp/protocol.py
+++ b/nbxmpp/protocol.py
@@ -936,7 +936,7 @@ class Protocol(Node):
         """
         self.setAttr('id', val)
 
-    def getError(self):
+    def getError(self) -> Optional[str]:
         """
         Return the error-condition (if present) or the textual description
         of the error (otherwise)
diff --git a/nbxmpp/task.py b/nbxmpp/task.py
index 56ef0e2..252a477 100644
--- a/nbxmpp/task.py
+++ b/nbxmpp/task.py
@@ -267,7 +267,7 @@ class Task:
             return self._error
         return self._result
 
-    def finish(self):
+    def finish(self) -> Any:
         if self._error is not None:
             raise self._error  # pylint: disable=raising-bad-type
         return self._result