Skip to content

Instantly share code, notes, and snippets.

@krsnewwave
Created March 11, 2022 15:10
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save krsnewwave/374460a92090e2b0ab991bdd1bdb01f4 to your computer and use it in GitHub Desktop.
Save krsnewwave/374460a92090e2b0ab991bdd1bdb01f4 to your computer and use it in GitHub Desktop.
kedro pipeline registry
# in <root>/src/<project>/pipeline_registry.py
def register_pipelines() -> Dict[str, Pipeline]:
data_engineering_pipeline = de.create_pipeline()
xgb_pipe = ds.create_xgb_pipeline()
rr_pipe = ds.create_rr_pipeline()
logres_pipe = ds.create_logres_pipeline()
rr_ho_pipe = ds.create_rr_ho_pipeline()
return {
"de": data_engineering_pipeline,
"xgb_pipe": xgb_pipe,
"rr_pipe": rr_pipe,
"rr_ho_pipe" : rr_ho_pipe,
"logres_pipe": logres_pipe,
"__default__": data_engineering_pipeline + xgb_pipe + rr_pipe + logres_pipe + rr_ho_pipe,
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment