1 files changed, 7 insertions, 9 deletions
diff --git a/sphinx/extension.py b/sphinx/extension.py
index 7565cbc85..3cab20c2f 100644
--- a/sphinx/extension.py
+++ b/sphinx/extension.py
@@ -8,22 +8,22 @@
     :license: BSD, see LICENSE for details.
 """
 
+from typing import Any, Dict
+
+from sphinx.config import Config
 from sphinx.errors import VersionRequirementError
 from sphinx.locale import __
 from sphinx.util import logging
 
 if False:
     # For type annotation
-    from typing import Any, Dict  # NOQA
-    from sphinx.application import Sphinx  # NOQA
-    from sphinx.config import Config  # NOQA
+    from sphinx.application import Sphinx
 
 logger = logging.getLogger(__name__)
 
 
 class Extension:
-    def __init__(self, name, module, **kwargs):
-        # type: (str, Any, Any) -> None
+    def __init__(self, name: str, module: Any, **kwargs) -> None:
         self.name = name
         self.module = module
         self.metadata = kwargs
@@ -40,8 +40,7 @@ class Extension:
         self.parallel_write_safe = kwargs.pop('parallel_write_safe', True)
 
 
-def verify_needs_extensions(app, config):
-    # type: (Sphinx, Config) -> None
+def verify_needs_extensions(app: "Sphinx", config: Config) -> None:
     """Verify the required Sphinx extensions are loaded."""
     if config.needs_extensions is None:
         return
@@ -60,8 +59,7 @@ def verify_needs_extensions(app, config):
                                           (extname, reqversion, extension.version))
 
 
-def setup(app):
-    # type: (Sphinx) -> Dict[str, Any]
+def setup(app: "Sphinx") -> Dict[str, Any]:
     app.connect('config-inited', verify_needs_extensions)
 
     return {