Make load_profile importable from aiida module

src/aiida/__init__.py

@@ -19,8 +19,15 @@
 More information at http://www.aiida.net
 """
 
-from aiida.common.log import configure_logging  # noqa: F401
-from aiida.manage.configuration import get_config_option, get_profile, load_profile, profile_context  # noqa: F401
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from aiida.common.log import configure_logging
+from aiida.manage.configuration import get_config_option, get_profile, load_profile, profile_context
+
+if TYPE_CHECKING:
+    from IPython.core.interactiveshell import InteractiveShell
 
 __copyright__ = (
     'Copyright (c), This file is part of the AiiDA platform. '
@@ -35,6 +42,8 @@
 )
 __paper_short__ = 'S. P. Huber et al., Scientific Data 7, 300 (2020).'
 
+__all__ = ['load_profile', 'configure_logging', 'get_config_option', 'get_profile', 'profile_context']
+
 
 def get_strict_version():
     """Return a distutils StrictVersion instance with the current distribution version
@@ -93,7 +102,7 @@
     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
 

src/aiida/common/log.py

@@ -17,6 +17,7 @@
 import logging
 import types
 import typing as t
+from pathlib import Path
 
 __all__ = ('AIIDA_LOGGER', 'override_log_level')
 
@@ -164,7 +165,7 @@ def evaluate_logging_configuration(dictionary):
     return result
 
 
-def configure_logging(with_orm=False, daemon=False, daemon_log_file=None):
+def configure_logging(with_orm: bool = False, daemon: bool = False, daemon_log_file: None | Path = None) -> None:
     """Setup the logging by retrieving the LOGGING dictionary from aiida and passing it to
     the python module logging.config.dictConfig. If the logging needs to be setup for the
     daemon, set the argument 'daemon' to True and specify the path to the log file. This