Skip to content

Commit

Permalink
revert debugging
Browse files Browse the repository at this point in the history
Signed-off-by: ravi-kumar-pilla <[email protected]>
  • Loading branch information
ravi-kumar-pilla committed Apr 3, 2024
1 parent 3aafecf commit 216dfbe
Showing 1 changed file with 1 addition and 22 deletions.
23 changes: 1 addition & 22 deletions package/kedro_viz/integrations/kedro/hooks.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,13 +6,12 @@
import logging
from collections import defaultdict
from pathlib import Path
from typing import Any, Dict, Union
from typing import Any, Union

from kedro.framework.hooks import hook_impl
from kedro.io import DataCatalog
from kedro.io.core import get_filepath_str
from kedro.pipeline.pipeline import TRANSCODING_SEPARATOR, _strip_transcoding
from kedro.runner.parallel_runner import ParallelRunner

logger = logging.getLogger(__name__)

Expand All @@ -39,26 +38,6 @@ def after_catalog_created(self, catalog: DataCatalog):
# Support for Kedro 0.18.x
self.datasets = catalog._data_sets # type: ignore[attr-defined]

@hook_impl
def before_pipeline_run(
self, run_params: Dict[str, Any], pipeline, catalog, **kwargs
) -> None:
"""Hook implementation to start an MLflow run
with the session_id of the Kedro pipeline run.
"""
import pdb

pdb.set_trace()

pipeline_runner_obj = run_params["runner"]
print(run_params)
if isinstance(pipeline_runner_obj, ParallelRunner):
print(pipeline_runner_obj._manager)
runnerManager = pipeline_runner_obj._manager
self._stats = runnerManager.dict()
else:
print("Sq Runner")

@hook_impl
def after_dataset_loaded(self, dataset_name: str, data: Any):
"""Hook to be invoked after a dataset is loaded from the catalog.
Expand Down

0 comments on commit 216dfbe

Please sign in to comment.