diff --git a/kedro/framework/session/session.py b/kedro/framework/session/session.py index 15bfad7800..6192793bce 100644 --- a/kedro/framework/session/session.py +++ b/kedro/framework/session/session.py @@ -77,8 +77,6 @@ def _jsonify_cli_context(ctx: click.core.Context) -> Dict[str, Any]: def _clear_hook_manager(): - from kedro.framework.hooks import get_hook_manager - hook_manager = get_hook_manager() name_plugin_pairs = hook_manager.list_name_plugin() for name, plugin in name_plugin_pairs: diff --git a/tests/framework/session/test_session_extension_hooks.py b/tests/framework/session/test_session_extension_hooks.py index 5f1b8c1473..2cb1f0b2c3 100644 --- a/tests/framework/session/test_session_extension_hooks.py +++ b/tests/framework/session/test_session_extension_hooks.py @@ -470,7 +470,9 @@ class MockSettings(_ProjectSettings): _HOOKS = Validator("HOOKS", default=(project_hooks, BeforeNodeRunHook())) _mock_imported_settings_paths(mocker, MockSettings()) - return KedroSession.create(mock_package_name, tmp_path) + session = KedroSession.create(mock_package_name, tmp_path) + yield session + session.close() class TestBeforeNodeRunHookWithInputUpdates: