diff options
| author | S. Solomon Darnell | 2025-03-28 21:52:21 -0500 |
|---|---|---|
| committer | S. Solomon Darnell | 2025-03-28 21:52:21 -0500 |
| commit | 4a52a71956a8d46fcb7294ac71734504bb09bcc2 (patch) | |
| tree | ee3dc5af3b6313e921cd920906356f5d4febc4ed /.venv/lib/python3.12/site-packages/azure/ai/ml/dsl/_mldesigner | |
| parent | cc961e04ba734dd72309fb548a2f97d67d578813 (diff) | |
| download | gn-ai-master.tar.gz | |
Diffstat (limited to '.venv/lib/python3.12/site-packages/azure/ai/ml/dsl/_mldesigner')
| -rw-r--r-- | .venv/lib/python3.12/site-packages/azure/ai/ml/dsl/_mldesigner/__init__.py | 46 | ||||
| -rw-r--r-- | .venv/lib/python3.12/site-packages/azure/ai/ml/dsl/_mldesigner/_constants.py | 33 |
2 files changed, 79 insertions, 0 deletions
diff --git a/.venv/lib/python3.12/site-packages/azure/ai/ml/dsl/_mldesigner/__init__.py b/.venv/lib/python3.12/site-packages/azure/ai/ml/dsl/_mldesigner/__init__.py new file mode 100644 index 00000000..2732cd75 --- /dev/null +++ b/.venv/lib/python3.12/site-packages/azure/ai/ml/dsl/_mldesigner/__init__.py @@ -0,0 +1,46 @@ +# --------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# --------------------------------------------------------- + +"""This file stores functions and objects that will be used in mldesigner package. + +DO NOT change the module names in "all" list. If the interface has changed in source code, wrap it here and keep +original function/module names the same as before, otherwise mldesigner will be broken by this change. +""" + +__path__ = __import__("pkgutil").extend_path(__path__, __name__) + +from azure.ai.ml._internal.entities import InternalComponent +from azure.ai.ml._internal.entities._additional_includes import InternalAdditionalIncludes +from azure.ai.ml._utils._asset_utils import get_ignore_file +from azure.ai.ml._utils.utils import try_enable_internal_components +from azure.ai.ml.dsl._condition import condition +from azure.ai.ml.dsl._do_while import do_while +from azure.ai.ml.dsl._group_decorator import group +from azure.ai.ml.dsl._parallel_for import ParallelFor, parallel_for +from azure.ai.ml.entities._component.component_factory import component_factory +from azure.ai.ml.entities._inputs_outputs import _get_param_with_standard_annotation +from azure.ai.ml.entities._job.pipeline._load_component import _generate_component_function + +from ._constants import V1_COMPONENT_TO_NODE + +component_factory_load_from_dict = component_factory.load_from_dict + + +__all__ = [ + # to be put in main package + "condition", + "do_while", + "group", + "parallel_for", + "ParallelFor", + # must keep + "get_ignore_file", + "_get_param_with_standard_annotation", + "_generate_component_function", + "component_factory_load_from_dict", + "V1_COMPONENT_TO_NODE", + "try_enable_internal_components", + "InternalAdditionalIncludes", + "InternalComponent", +] diff --git a/.venv/lib/python3.12/site-packages/azure/ai/ml/dsl/_mldesigner/_constants.py b/.venv/lib/python3.12/site-packages/azure/ai/ml/dsl/_mldesigner/_constants.py new file mode 100644 index 00000000..f770c97d --- /dev/null +++ b/.venv/lib/python3.12/site-packages/azure/ai/ml/dsl/_mldesigner/_constants.py @@ -0,0 +1,33 @@ +# --------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# --------------------------------------------------------- + +"""This file stores constants that will be used in mldesigner package.""" +from azure.ai.ml._internal._schema.component import NodeType as V1NodeType +from azure.ai.ml._internal.entities import ( + Ae365exepool, + AetherBridge, + Command as InternalCommand, + Parallel as InternalParallel, + Pipeline as InternalPipeline, + DataTransfer, + Distributed, + HDInsight, + Hemera, + Scope, + Starlite, +) + +V1_COMPONENT_TO_NODE = { + V1NodeType.SCOPE: Scope, + V1NodeType.COMMAND: InternalCommand, + V1NodeType.PARALLEL: InternalParallel, + V1NodeType.PIPELINE: InternalPipeline, + V1NodeType.DATA_TRANSFER: DataTransfer, + V1NodeType.DISTRIBUTED: Distributed, + V1NodeType.HDI: HDInsight, + V1NodeType.STARLITE: Starlite, + V1NodeType.HEMERA: Hemera, + V1NodeType.AE365EXEPOOL: Ae365exepool, + V1NodeType.AETHER_BRIDGE: AetherBridge, +} |
