Add support for %load_ext kedro

RELEASE.md

@@ -4,6 +4,7 @@
 * Implemented `KedroDataCatalog.to_config()` method that converts the catalog instance into a configuration format suitable for serialization.
 * Improve OmegaConfigLoader performance.
 * Replaced `trufflehog` with `detect-secrets` for detecting secrets within a code base.
+* Added support for `%load_ext kedro`.
 
 ## Bug fixes and other changes
 * Added validation to ensure dataset versions consistency across catalog.

kedro/__init__.py

@@ -5,6 +5,7 @@
 
 import sys
 import warnings
+from typing import Any
 
 __version__ = "0.19.10"
 
@@ -29,3 +30,9 @@ class KedroPythonVersionWarning(UserWarning):
 or set the PYTHONWARNINGS environment variable accordingly.""",
         KedroPythonVersionWarning,
     )
+
+
+def load_ipython_extension(ipython: Any) -> None:
+    import kedro.ipython
+
+    kedro.ipython.load_ipython_extension(ipython)

pyproject.toml

@@ -178,7 +178,8 @@ layers = [
 ]
 ignore_imports = [
     "kedro.runner.task -> kedro.framework.project",
-    "kedro.framework.hooks.specs -> kedro.framework.context"
+    "kedro.framework.hooks.specs -> kedro.framework.context",
+    "kedro -> kedro.ipython"
 ]
 
 [[tool.importlinter.contracts]]
@@ -188,6 +189,9 @@ modules = [
     "kedro.pipeline",
     "kedro.io"
 ]
+ignore_imports = [
+    "kedro -> kedro.ipython"
+]
 
 [[tool.importlinter.contracts]]
 name = "Config cannot import Runner et al"

tests/ipython/test_ipython.py

@@ -222,6 +222,17 @@ def test_line_magic_with_invalid_arguments(self, mocker, ipython):
         ):
             ipython.magic("reload_kedro --invalid_arg=dummy")
 
+    def test_ipython_kedro_extension_alias(self, mocker, ipython):
+        mock_ipython_extension = mocker.patch(
+            "kedro.ipython.load_ipython_extension", autospec=True
+        )
+        # Ensure that `kedro` is not loaded initially
+        assert "kedro" not in ipython.extension_manager.loaded
+        ipython.magic("load_ext kedro")
+        mock_ipython_extension.assert_called_once_with(ipython)
+        # Ensure that `kedro` extension has been loaded
+        assert "kedro" in ipython.extension_manager.loaded
+
 
 class TestProjectPathResolution:
     def test_only_path_specified(self):