Skip to content

Commit 87a8696

Browse files
authored
Add services to finetune and distillation jobs (Azure#38710)
1 parent 1726b0c commit 87a8696

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

sdk/ml/azure-ai-ml/azure/ai/ml/entities/_job/distillation/distillation_job.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -334,6 +334,7 @@ def _from_rest_object(cls, obj: RestJobBase) -> "DistillationJob":
334334
"tags": properties.tags,
335335
"properties": properties.properties,
336336
"experiment_name": properties.experiment_name,
337+
"services": properties.services,
337338
"status": properties.status,
338339
"creation_context": obj.system_data,
339340
"display_name": properties.display_name,
@@ -381,6 +382,7 @@ def _to_rest_object(self) -> "RestFineTuningJob":
381382
display_name=self.display_name,
382383
description=self.description,
383384
experiment_name=self.experiment_name,
385+
services=self.services,
384386
tags=self.tags,
385387
properties=self.properties,
386388
fine_tuning_details=distillation,

sdk/ml/azure-ai-ml/azure/ai/ml/entities/_job/finetuning/custom_model_finetuning_job.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -87,6 +87,7 @@ def _to_rest_object(self) -> "RestFineTuningJob":
8787
display_name=self.display_name,
8888
description=self.description,
8989
experiment_name=self.experiment_name,
90+
services=self.services,
9091
tags=self.tags,
9192
properties=self.properties,
9293
fine_tuning_details=custom_finetuning_vertical,
@@ -162,6 +163,7 @@ def _from_rest_object(cls, obj: RestJobBase) -> "CustomModelFineTuningJob":
162163
"description": properties.description,
163164
"tags": properties.tags,
164165
"properties": properties.properties,
166+
"services": properties.services,
165167
"experiment_name": properties.experiment_name,
166168
"status": properties.status,
167169
"creation_context": obj.system_data,

0 commit comments

Comments
 (0)