Enforce .encode() return type using mypy (#551)

Closes #505

2 files changed, 2 insertions, 2 deletions

diff --git a/jwt/api_jws.py b/jwt/api_jws.py
index f01f53e..bce7a1a 100644
--- a/jwt/api_jws.py
+++ b/jwt/api_jws.py
@@ -79,7 +79,7 @@ class PyJWS:
         algorithm: str = "HS256",
         headers: Optional[Dict] = None,
         json_encoder: Optional[Type[json.JSONEncoder]] = None,
-    ):
+    ) -> str:
         segments = []
 
         if algorithm is None:
diff --git a/jwt/api_jwt.py b/jwt/api_jwt.py
index d5bcb15..60f319b 100644
--- a/jwt/api_jwt.py
+++ b/jwt/api_jwt.py
@@ -40,7 +40,7 @@ class PyJWT(PyJWS):
         algorithm: str = "HS256",
         headers: Optional[Dict] = None,
         json_encoder: Optional[Type[json.JSONEncoder]] = None,
-    ):
+    ) -> str:
         # Check that we get a mapping
         if not isinstance(payload, Mapping):
             raise TypeError(