Skip to content

Commit

Permalink
remove session store from _load_data_helper
Browse files Browse the repository at this point in the history
Signed-off-by: Huong Nguyen <[email protected]>
  • Loading branch information
Huong Nguyen committed Dec 3, 2024
1 parent ab66d10 commit 0490962
Showing 1 changed file with 2 additions and 4 deletions.
6 changes: 2 additions & 4 deletions package/kedro_viz/integrations/kedro/data_loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -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(
Expand All @@ -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
Expand All @@ -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(
Expand Down

0 comments on commit 0490962

Please sign in to comment.