Close session in fixture

kedro-org · Jan 13, 2022 · cf8979a · cf8979a
1 parent e5393b3
commit cf8979a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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: