add experimental decorator for datatransfer (#29562)

Co-authored-by: Clement Wang <clwan@microsoft.com>
Azure · Mar 24, 2023 · f02be13 · f02be13
1 parent eae88f0
commit f02be13
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 0 deletions.
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_builders/data_transfer.py b/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_builders/data_transfer.py
@@ -15,6 +15,7 @@
     DataTransferImportJobSchema,
     DataTransferExportJobSchema,
 )
+from azure.ai.ml._utils._experimental import experimental
 from azure.ai.ml.constants._component import (
     NodeType,
     ExternalDataType,
@@ -147,6 +148,7 @@ def _build_inputs(self):
         return built_inputs
 
 
+@experimental
 class DataTransferCopy(DataTransfer):
     """Base class for data transfer copy node.
 
@@ -290,6 +292,7 @@ def __call__(self, *args, **kwargs) -> "DataTransferCopy":
         )
 
 
+@experimental
 class DataTransferImport(DataTransfer):
     """Base class for data transfer import node.
 
@@ -423,6 +426,7 @@ def _to_job(self) -> DataTransferImportJob:
         )
 
 
+@experimental
 class DataTransferExport(DataTransfer):
     """Base class for data transfer export node.
 

diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_builders/data_transfer_func.py b/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_builders/data_transfer_func.py
@@ -5,6 +5,7 @@
 
 from typing import Optional, Dict, Union, Callable, Tuple
 
+from azure.ai.ml._utils._experimental import experimental
 from azure.ai.ml.entities._component.datatransfer_component import (
     DataTransferCopyComponent,
 )
@@ -117,6 +118,7 @@ def _parse_inputs_outputs(io_dict: Dict, parse_func: Callable) -> Tuple[Dict, Di
     return component_io_dict, job_io_dict
 
 
+@experimental
 def copy_data(
     *,
     name: Optional[str] = None,
@@ -194,6 +196,7 @@ def copy_data(
     return data_transfer_copy_obj
 
 
+@experimental
 @pipeline_node_decorator
 def import_data(
     *,
@@ -259,6 +262,7 @@ def import_data(
     return data_transfer_import_obj
 
 
+@experimental
 @pipeline_node_decorator
 def export_data(
     *,

diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_component/datatransfer_component.py b/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_component/datatransfer_component.py
@@ -11,6 +11,7 @@
     DataTransferImportComponentSchema,
     DataTransferExportComponentSchema,
 )
+from azure.ai.ml._utils._experimental import experimental
 from azure.ai.ml.constants._common import COMPONENT_TYPE, AssetTypes
 from azure.ai.ml.constants._component import (
     NodeType,
@@ -130,6 +131,7 @@ def _build_source_sink(cls, io_dict: Union[Dict, Database, FileSystem]):
         return component_io
 
 
+@experimental
 class DataTransferCopyComponent(DataTransferComponent):
     """DataTransfer copy component version, used to define a data transfer copy component.
 
@@ -222,6 +224,7 @@ def _validate_input_output_mapping(self):
         return validation_result
 
 
+@experimental
 class DataTransferImportComponent(DataTransferComponent):
     """DataTransfer import component version, used to define a data transfer import component.
 
@@ -266,6 +269,7 @@ def __call__(self, *args, **kwargs):
         )
 
 
+@experimental
 class DataTransferExportComponent(DataTransferComponent):  # pylint: disable=too-many-instance-attributes
     """DataTransfer export component version, used to define a data transfer export component.