Commit d51f2ba5 authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

add dump_dfs_plan

parent 295e39a5
Pipeline #11832 passed with stage
in 0 seconds
...@@ -258,6 +258,12 @@ class Pipeline: ...@@ -258,6 +258,12 @@ class Pipeline:
print(ccds_refs_abs) print(ccds_refs_abs)
return ccds_refs_abs return ccds_refs_abs
def dump_dfs_plan(self, dfs_plan):
self.mkdir(os.path.join(self.dir_output, "dfs"))
output_path = os.path.join(self.dir_output, "dfs", "dfs_plan.joblib")
joblib.dump(dfs_plan, output_path)
return output_path
def dump_dfs_recs( def dump_dfs_recs(
self, dfs_recs_abs: list[dict], dir_dump: str = None self, dfs_recs_abs: list[dict], dir_dump: str = None
) -> list[dict]: ) -> list[dict]:
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment