Revert "[ML] Removed Experimental Tags from OS Image Metadata (Azure#…

…29158)" This reverts commit b393611.
nthandeMS · Mar 16, 2023 · 65b3c79 · 65b3c79
1 parent 4ede540
commit 65b3c79
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 2 deletions.
diff --git a/sdk/ml/azure-ai-ml/CHANGELOG.md b/sdk/ml/azure-ai-ml/CHANGELOG.md
@@ -6,7 +6,6 @@
 - Added support for `tags` on Compute Resources.
 - Added support for promoting data asset from a workspace to a registry
 - Added support for registering named asset from job output or node output by specifying name and version settings.
-- Removed Experimental Tag from Image Metadata on Compute Instances.
 - Added support for data binding on outputs inside dynamic arguments for dsl pipeline
 - Added support for serverless compute in pipeline, command, automl and sweep job
 - Added support for `job_tier` and `priority` in job

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
@@ -67,7 +67,7 @@ class ComputeInstanceSchema(ComputeSchema):
     idle_time_before_shutdown_minutes = ExperimentalField(fields.Int())
     custom_applications = ExperimentalField(fields.List(NestedField(CustomApplicationsSchema)))
     setup_scripts = ExperimentalField(NestedField(SetupScriptsSchema))
-    os_image_metadata = NestedField(OsImageMetadataSchema, dump_only=True)
+    os_image_metadata = ExperimentalField(NestedField(OsImageMetadataSchema, dump_only=True))
     enable_node_public_ip = fields.Bool(
         metadata={"description": "Enable or disable node public IP address provisioning."}
     )
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_compute/_image_metadata.py b/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_compute/_image_metadata.py
@@ -1,8 +1,10 @@
 # ---------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # ---------------------------------------------------------
+from azure.ai.ml._utils._experimental import experimental
 
 
+@experimental
 class ImageMetadata:
     """Metadata about the operating system image for this compute instance."""
 

diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_compute/compute_instance.py b/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_compute/compute_instance.py
@@ -22,6 +22,7 @@
 )
 from azure.ai.ml._schema._utils.utils import get_subnet_str
 from azure.ai.ml._schema.compute.compute_instance import ComputeInstanceSchema
+from azure.ai.ml._utils._experimental import experimental
 from azure.ai.ml.constants._common import BASE_PATH_CONTEXT_KEY, TYPE
 from azure.ai.ml.constants._compute import ComputeDefaults, ComputeType
 from azure.ai.ml.entities._compute.compute import Compute, NetworkSettings
@@ -223,6 +224,7 @@ def state(self) -> str:
         """
         return self._state
 
+    @experimental
     @property
     def os_image_metadata(self) -> ImageMetadata:
         """Metadata about the operating system image for this compute instance.