Skip to content

Commit b765c1f

Browse files
authored
Added deployment type in telemetry. (#1291)
2 parents 3b49d21 + de60899 commit b765c1f

File tree

2 files changed

+8
-0
lines changed

2 files changed

+8
-0
lines changed

ads/aqua/modeldeployment/constants.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,5 +16,6 @@
1616

1717

1818
class DeploymentType(ExtendedEnum):
19+
SINGLE = "SINGLE"
1920
STACKED = "STACKED"
2021
MULTI = "MULTI"

ads/aqua/modeldeployment/deployment.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1214,6 +1214,13 @@ def _create_deployment(
12141214
if Tags.BASE_MODEL_CUSTOM in tags:
12151215
telemetry_kwargs["custom_base_model"] = True
12161216

1217+
if Tags.MULTIMODEL_TYPE_TAG in tags:
1218+
telemetry_kwargs["deployment_type"] = DeploymentType.MULTI
1219+
elif Tags.STACKED_MODEL_TYPE_TAG in tags:
1220+
telemetry_kwargs["deployment_type"] = DeploymentType.STACKED
1221+
else:
1222+
telemetry_kwargs["deployment_type"] = DeploymentType.SINGLE
1223+
12171224
# tracks unique deployments that were created in the user compartment
12181225
self.telemetry.record_event_async(
12191226
category=f"aqua/{model_type}/deployment",

0 commit comments

Comments
 (0)