diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_datastore/one_lake.py b/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_datastore/one_lake.py index c3d65d301ef3..0d5a423d1eec 100644 --- a/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_datastore/one_lake.py +++ b/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_datastore/one_lake.py @@ -22,8 +22,10 @@ from azure.ai.ml.entities._datastore.utils import from_rest_datastore_credentials from azure.ai.ml.entities._mixins import DictMixin, RestTranslatableMixin from azure.ai.ml.entities._util import load_from_dict +from azure.ai.ml._utils._experimental import experimental +@experimental class OneLakeArtifact(RestTranslatableMixin, DictMixin, ABC): """OneLake artifact (data source) backing the OneLake workspace. @@ -39,6 +41,7 @@ def __init__(self, name: str, type: Optional[str] = None): self.type = type +@experimental class LakeHouseArtifact(OneLakeArtifact): """LakeHouse artifact type for OneLake. @@ -53,6 +56,7 @@ def _to_datastore_rest_object(self) -> RestLakeHouseArtifact: return RestLakeHouseArtifact(artifact_name=self.name) +@experimental class OneLakeDatastore(Datastore): """OneLake datastore that is linked to an Azure ML workspace.