Type for ipython var

src/aiida/__init__.py

@@ -20,6 +20,10 @@
 """
 
 from aiida.manage.configuration import load_profile
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from IPython.core.interactiveshell import InteractiveShell
 
 __copyright__ = (
     'Copyright (c), This file is part of the AiiDA platform. '
@@ -94,7 +98,7 @@ def get_file_header(comment_char: str = '# ') -> str:
     return '\n'.join(f'{comment_char}{line}' for line in lines)
 
 
-def load_ipython_extension(ipython):
+def load_ipython_extension(ipython: InteractiveShell):
     """Load the AiiDA IPython extension, using ``%load_ext aiida``."""
     from .tools.ipython.ipython_magics import AiiDALoaderMagics