Skip to content

Commit faaf7f5

Browse files
committed
Update category for telemetry model and shape used in deployment and finetune.
- added finetune shape, finetune model and combination of a shape and model
1 parent 4bbb213 commit faaf7f5

File tree

1 file changed

+11
-4
lines changed

1 file changed

+11
-4
lines changed

ads/aqua/finetune.py

Lines changed: 11 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -460,18 +460,25 @@ def create(
460460
telemetry_kwargs = (
461461
{"ocid": ft_job.id[-6:]} if ft_job and len(ft_job.id) > 6 else {}
462462
)
463+
# track shapes that were used for fine-tune creation
463464
self.telemetry.record_event_async(
464-
category="aqua/service/finetune/create",
465-
action="shape",
466-
detail=f"{create_fine_tuning_details.shape_name}x{create_fine_tuning_details.replica}",
467-
value=source.display_name,
465+
category=f"aqua/service/finetune/create/shape/",
466+
action=f"{create_fine_tuning_details.shape_name}x{create_fine_tuning_details.replica}",
468467
**telemetry_kwargs,
469468
)
470469
# tracks unique fine-tuned models that were created in the user compartment
471470
self.telemetry.record_event_async(
472471
category="aqua/service/finetune",
473472
action="create",
474473
detail=source.display_name,
474+
**telemetry_kwargs,
475+
)
476+
# track combination of model and shape used for fine-tune creation
477+
self.telemetry.record_event_async(
478+
category="aqua/service/finetune/create",
479+
action="shape",
480+
detail=f"{create_fine_tuning_details.shape_name}x{create_fine_tuning_details.replica}",
481+
value=source.display_name,
475482
)
476483

477484
return AquaFineTuningSummary(

0 commit comments

Comments
 (0)