Skip to content

Commit d0ac79f

Browse files
committed
feat: add deployment data to pipeline results
1 parent f60fcb4 commit d0ac79f

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

ami/ml/models/pipeline.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -911,6 +911,7 @@ class PipelineSaveResults:
911911
detections: list[Detection]
912912
classifications: list[Classification]
913913
algorithms: dict[str, Algorithm]
914+
deployments: dict[str, Deployment]
914915
total_time: float
915916

916917

@@ -964,14 +965,15 @@ def save_results(
964965

965966
deployments_data = results.deployments
966967
source_images_data = results.source_images
968+
deployments_used: dict[str, Deployment] = {}
967969

968970
if not deployments_data:
969971
job_logger.warning(
970972
"No deployments data found in results. "
971973
"New source images will not be created without deployments data."
972974
)
973975
else:
974-
get_or_create_deployments(
976+
deployments_used = get_or_create_deployments(
975977
deployments_data=deployments_data,
976978
project_id=project_id,
977979
logger=job_logger,
@@ -1099,6 +1101,7 @@ def save_results(
10991101
detections=detections,
11001102
classifications=classifications,
11011103
algorithms=algorithms_used,
1104+
deployments=deployments_used,
11021105
total_time=total_time,
11031106
)
11041107

0 commit comments

Comments
 (0)