diff --git a/sdk/ml/azure-ai-ml/CHANGELOG.md b/sdk/ml/azure-ai-ml/CHANGELOG.md index 064884a7edf8..25fd40d0d81c 100644 --- a/sdk/ml/azure-ai-ml/CHANGELOG.md +++ b/sdk/ml/azure-ai-ml/CHANGELOG.md @@ -13,7 +13,8 @@ ### Bugs Fixed ### Other Changes - - Removed declaration on Python 3.6 support. + - Removed declaration on Python 3.6 support + - Added support for custom setup scripts on compute instances. - Updated dependencies upper bounds to be major versions. ## 0.1.0b7 (In progress) diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/compute/compute_instance.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/compute/compute_instance.py index 92bd1c231db1..6e1c04523443 100644 --- a/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/compute/compute_instance.py +++ b/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/compute/compute_instance.py @@ -48,7 +48,7 @@ class ComputeInstanceSchema(ComputeSchema): state = fields.Str(dump_only=True) last_operation = fields.Dict(keys=fields.Str(), values=fields.Str(), dump_only=True) services = fields.List(fields.Dict(keys=fields.Str(), values=fields.Str()), dump_only=True) - schedules = ExperimentalField(NestedField(ComputeSchedulesSchema)) + schedules = NestedField(ComputeSchedulesSchema) identity = ExperimentalField(NestedField(IdentitySchema)) idle_time_before_shutdown = ExperimentalField(fields.Str()) setup_scripts = ExperimentalField(NestedField(SetupScriptsSchema)) diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/compute/setup_scripts.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/compute/setup_scripts.py index 9bf69fbc8f26..a4dbcad4c133 100644 --- a/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/compute/setup_scripts.py +++ b/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/compute/setup_scripts.py @@ -5,9 +5,8 @@ from marshmallow import fields from marshmallow.decorators import post_load -from azure.ai.ml._schema.core.fields import ArmVersionedStr, CodeField, LocalPathField, NestedField, UnionField +from azure.ai.ml._schema.core.fields import NestedField from azure.ai.ml._schema.core.schema_meta import PatchedSchemaMeta -from azure.ai.ml.constants._common import LOCAL_PATH, AzureMLResourceType class ScriptReferenceSchema(metaclass=PatchedSchemaMeta): diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_compute/_schedule.py b/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_compute/_schedule.py index 69e789c20600..96aee073d1cf 100644 --- a/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_compute/_schedule.py +++ b/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_compute/_schedule.py @@ -102,7 +102,6 @@ def _from_rest_object(cls, obj: RestComputeStartStopSchedule) -> "ComputeStartSt return schedule -@experimental class ComputeSchedules(RestTranslatableMixin): """Compute schedules.