rda

aiidateam · Dec 2, 2024 · 8bb411f · 8bb411f
1 parent 77ad915
commit 8bb411f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/aiida/__init__.py b/src/aiida/__init__.py
@@ -106,7 +106,10 @@ def get_file_header(comment_char: str = '# ') -> str:
 
 
 def load_ipython_extension(ipython):
-    """Load the AiiDA IPython extension, using ``%load_ext aiida``."""
+    """Load the AiiDA IPython extension, using ``%load_ext aiida``.
+
+    :param ipython: InteractiveShell instance. If ``None``, the global InteractiveShell is used.
+    """
     from aiida.tools.ipython.ipython_magics import load_ipython_extension
 
     load_ipython_extension(ipython)
diff --git a/src/aiida/tools/ipython/ipython_magics.py b/src/aiida/tools/ipython/ipython_magics.py
@@ -205,7 +205,6 @@ def register_ipython_extension(ipython=None):
     The %aiida IPython extension provides the same environment as the `verdi shell`.
 
     :param ipython: InteractiveShell instance. If omitted, the global InteractiveShell is used.
-
     """
     if ipython is None:
         ipython = get_ipython()