1 files changed, 3 insertions, 7 deletions
diff --git a/sphinx/ext/jsmath.py b/sphinx/ext/jsmath.py
index 47e1b1836..84f191a5e 100644
--- a/sphinx/ext/jsmath.py
+++ b/sphinx/ext/jsmath.py
@@ -10,6 +10,7 @@
 """
 
 import warnings
+from typing import Any, Dict
 
 from sphinxcontrib.jsmath import (  # NOQA
     html_visit_math,
@@ -18,16 +19,11 @@ from sphinxcontrib.jsmath import (  # NOQA
 )
 
 import sphinx
+from sphinx.application import Sphinx
 from sphinx.deprecation import RemovedInSphinx40Warning
 
-if False:
-    # For type annotation
-    from typing import Any, Dict  # NOQA
-    from sphinx.application import Sphinx  # NOQA
 
-
-def setup(app):
-    # type: (Sphinx) -> Dict[str, Any]
+def setup(app: Sphinx) -> Dict[str, Any]:
     warnings.warn('sphinx.ext.jsmath has been moved to sphinxcontrib-jsmath.',
                   RemovedInSphinx40Warning)