Skip to content

Commit 7e70f4b

Browse files
authored
fix: pass attribute aether for AetherBridgeComponent (#35534)
1 parent 3a2c666 commit 7e70f4b

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

sdk/ml/azure-ai-ml/azure/ai/ml/_internal/entities/component.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -103,6 +103,7 @@ def __init__(
103103
ae365exepool: Optional[Dict] = None,
104104
launcher: Optional[Dict] = None,
105105
datatransfer: Optional[Dict] = None,
106+
aether: Optional[Dict] = None,
106107
**kwargs,
107108
):
108109
_type, self._type_label = parse_name_label(type)
@@ -141,6 +142,7 @@ def __init__(
141142
self.ae365exepool = ae365exepool
142143
self.launcher = launcher
143144
self.datatransfer = datatransfer
145+
self.aether = aether
144146

145147
@classmethod
146148
def _build_io(cls, io_dict: Union[Dict, Input, Output], is_input: bool):

sdk/ml/azure-ai-ml/tests/internal/unittests/test_component.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -989,6 +989,7 @@ def test_load_from_internal_aether_bridge_component(self):
989989
"mock_param1": {"type": "AnyFile", "optional": False},
990990
"mock_param2": {"type": "AnyFile", "optional": False},
991991
},
992+
"aether": {"module_type": "ScrapingCloud", "ref_id": "mock_ref_id"},
992993
"outputs": {"job_info": {"type": "AnyFile"}},
993994
"type": "AetherBridgeComponent",
994995
"command": "mock.exe {inputs.mock_param1} {inputs.mock_param2} {outputs.job_info}",

0 commit comments

Comments
 (0)