Skip to content

Commit 9242b45

Browse files
authored
[ML][Pipelines] Update curated environment for expression component (Azure#28908)
* update expression environment to mldesigner-minimal * update expression environment to mldesigner * enable private preview features for expression component
1 parent 10a2303 commit 9242b45

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

sdk/ml/azure-ai-ml/azure/ai/ml/entities/_job/pipeline/_pipeline_expression.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
from typing import Dict, List, Tuple, Union
1212

1313
from azure.ai.ml._utils.utils import dump_yaml_to_file, get_all_data_binding_expressions, load_yaml
14+
from azure.ai.ml.constants._common import AZUREML_PRIVATE_FEATURES_ENV_VAR
1415
from azure.ai.ml.constants._component import ComponentParameterTypes, IOConstants
1516
from azure.ai.ml.exceptions import UserErrorException
1617

@@ -577,4 +578,5 @@ def _generate_yaml_file(_path: Path) -> None:
577578
component_func = load_component(yaml_path)
578579
component_kwargs = {k: v.value for k, v in self._inputs.items()}
579580
self._created_component = component_func(**component_kwargs)
581+
self._created_component.environment_variables = {AZUREML_PRIVATE_FEATURES_ENV_VAR: "true"}
580582
return self._created_component

sdk/ml/azure-ai-ml/azure/ai/ml/entities/_job/pipeline/data/expression_component_template.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ outputs:
1010

1111
code: ./src
1212

13-
environment: azureml:AzureML-MLDesigner@latest
13+
environment: azureml://registries/azureml/environments/mldesigner/labels/latest
1414

1515
command: >-
1616
{command_placeholder}

0 commit comments

Comments
 (0)