From 0490962469383db1ff3896308ce17e2e79ddd6ca Mon Sep 17 00:00:00 2001 From: Huong Nguyen Date: Tue, 3 Dec 2024 15:11:33 +0000 Subject: [PATCH] remove session store from _load_data_helper Signed-off-by: Huong Nguyen --- package/kedro_viz/integrations/kedro/data_loader.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/package/kedro_viz/integrations/kedro/data_loader.py b/package/kedro_viz/integrations/kedro/data_loader.py index fc34abcb3..ec2d6671b 100644 --- a/package/kedro_viz/integrations/kedro/data_loader.py +++ b/package/kedro_viz/integrations/kedro/data_loader.py @@ -72,8 +72,7 @@ def _load_data_helper( configuration. is_lite: A flag to run Kedro-Viz in lite mode. Returns: - A tuple containing the data catalog, pipeline dictionary, session store - and dataset stats dictionary. + A tuple containing the data catalog, pipeline dictionary and dataset stats dictionary. """ with KedroSession.create( @@ -87,7 +86,6 @@ def _load_data_helper( session._hook_manager = _VizNullPluginManager() # type: ignore context = session.load_context() - session_store = session._store # patch the AbstractDataset class for a custom # implementation to handle kedro.io.core.DatasetError @@ -109,7 +107,7 @@ def _load_data_helper( # Useful for users who have `get_current_session` in their `register_pipelines()`. pipelines_dict = dict(pipelines) stats_dict = _get_dataset_stats(project_path) - return catalog, pipelines_dict, session_store, stats_dict + return catalog, pipelines_dict, stats_dict def load_data(