Skip to content

Commit a1d3429

Browse files
fix(workflows): update combine time fate correlation task output directory
Signed-off-by: Cameron Smith <cameron.ray.smith@gmail.com>
1 parent c9d90b4 commit a1d3429

File tree

1 file changed

+5
-6
lines changed

1 file changed

+5
-6
lines changed

src/pyrovelocity/workflows/main_workflow.py

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -506,14 +506,13 @@ def map_model_configurations_over_data_set(
506506
def combine_time_lineage_fate_correlation(
507507
results: List[List[SummarizeOutputs]],
508508
) -> List[FlyteFile]:
509-
print(results)
509+
output_dir = Path("reports/time_fate_correlation")
510+
output_dir.mkdir(parents=True, exist_ok=True)
511+
510512
model_ordered_results = list(map(list, zip(*results)))
511-
print(model_ordered_results)
512513
time_lineage_fate_correlation_plots = []
513514

514515
for model_results in model_ordered_results:
515-
print(model_results)
516-
517516
prepared_model_results = []
518517
for model_output in model_results:
519518
postprocessed_data_path = model_output.postprocessed_data.download()
@@ -536,7 +535,7 @@ def combine_time_lineage_fate_correlation(
536535
"Multilineage",
537536
"All lineages",
538537
][: len(prepared_model_results)],
539-
reports_path=Path("."),
538+
output_dir=output_dir,
540539
)
541540
)
542541

@@ -554,7 +553,7 @@ def combine_time_lineage_fate_correlation(
554553
upload_result = upload_file_concurrently(
555554
bucket_name=f"pyrovelocity/reports/{execution_id}/time_fate_correlation",
556555
source_filename=f"{file}{ext}",
557-
destination_blob_name=f"{file}{ext}",
556+
destination_blob_name=f"{file.name}{ext}",
558557
)
559558
attempted_upload_results.append(upload_result)
560559

0 commit comments

Comments
 (0)