more freshup

1 files changed, 4 insertions, 4 deletions

diff --git a/src/couch/feedreader.py b/src/couch/feedreader.py
index 98401ab..aac51d3 100644
--- a/src/couch/feedreader.py
+++ b/src/couch/feedreader.py
@@ -11,7 +11,7 @@ class BaseFeedReader:
         self.db = db
         return self
 
-    def on_message(self, message):
+    def on_message(self, message: str) -> None:
         """
         Callback method that is called when change
         message is received from couchdb.
@@ -22,14 +22,14 @@ class BaseFeedReader:
 
         raise NotImplementedError()
 
-    def on_close(self):
+    def on_close(self) -> None:
         """
         Callback method that is received when connection
         is closed with a server. By default, does nothing.
         """
         pass
 
-    def on_heartbeat(self):
+    def on_heartbeat(self) -> None:
         """
         Callback method invoked when a hearbeat (empty line) is received
         from the _changes stream. Override this to purge the reader's internal
@@ -48,5 +48,5 @@ class SimpleFeedReader(BaseFeedReader):
         self.callback = callback
         return super(SimpleFeedReader, self).__call__(db)
 
-    def on_message(self, message) -> None:
+    def on_message(self, message: str) -> None:
         self.callback(message, db=self.db)