Skip to content

Commit

Permalink
Remove redundant function from hooks
Browse files Browse the repository at this point in the history
Signed-off-by: Laura Couto <[email protected]>
  • Loading branch information
lrcouto committed Sep 19, 2024
1 parent bd16556 commit 6c5ac73
Showing 1 changed file with 0 additions and 13 deletions.
13 changes: 0 additions & 13 deletions performance-test/src/performance_test/hooks.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,6 @@
from time import sleep
from typing import Dict

from kedro.framework.hooks import hook_impl
from kedro.pipeline import Pipeline
from pyspark import SparkConf
from pyspark.sql import SparkSession

Expand All @@ -27,14 +25,3 @@ def after_context_created(self, context) -> None:
sleep(context.params['hook_delay'])
_spark_session = spark_session_conf.getOrCreate()
_spark_session.sparkContext.setLogLevel("WARN")


def register_pipelines(self) -> Dict[str, Pipeline]:
from performance_test.pipelines.expense_analysis import (
pipeline as expense_analysis_pipeline,
)

return {
"__default__": expense_analysis_pipeline.create_pipeline(),
"expense_analysis": expense_analysis_pipeline.create_pipeline(),
}

0 comments on commit 6c5ac73

Please sign in to comment.