Adding machine learning extension

src/machinelearningservices/HISTORY.rst

@@ -0,0 +1,8 @@
+.. :changelog:
+
+Release History
+===============
+
+2.38.0
+++++++
+* Fix a bug compute update which caused Enable SSO property to reset.

src/machinelearningservices/README.rst

@@ -0,0 +1,20 @@
+Azure CLI ml extension
+==============================================================================
+
+.. contents::
+    :depth: 1
+    :local:
+
+Overview
+------------------------------------------------------------------------------
+This is the extension for ml
+
+
+Get started with Azure CLI ml extension
+------------------------------------------------------------------------------
+
+Install this extension using the below CLI command:
+
+    .. code-block:: bash
+
+        az extension add --name ml

src/machinelearningservices/azext_mlv2/__init__.py

@@ -0,0 +1,55 @@
+# --------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for
+# license information.
+#
+# Code generated by Microsoft (R) AutoRest Code Generator.
+# Changes may cause incorrect behavior and will be lost if the code is
+# regenerated.
+# --------------------------------------------------------------------------
+
+from azure.cli.core import AzCommandsLoader
+
+from azext_mlv2.generated._help import helps  # pylint: disable=unused-import
+
+try:
+    from azext_mlv2.manual._help import helps  # pylint: disable=reimported
+except ImportError as e:
+    print(e)
+
+
+class AzureMachineLearningWorkspacesCommandsLoader(AzCommandsLoader):
+    def __init__(self, cli_ctx=None):
+        from azext_mlv2.generated._client_factory import cf_ml_cl
+
+        from azure.cli.core.commands import CliCommandType
+
+        ml_custom = CliCommandType(operations_tmpl="azext_mlv2.custom#{}", client_factory=cf_ml_cl)
+        parent = super(AzureMachineLearningWorkspacesCommandsLoader, self)
+        parent.__init__(cli_ctx=cli_ctx, custom_command_type=ml_custom)
+
+    def load_command_table(self, args):
+        from azext_mlv2.generated.commands import load_command_table
+
+        load_command_table(self, args)
+        try:
+            from azext_mlv2.manual.commands import load_command_table as load_command_table_manual
+
+            load_command_table_manual(self, args)
+        except ImportError:
+            pass
+        return self.command_table
+
+    def load_arguments(self, command):
+        from azext_mlv2.generated._params import load_arguments
+
+        load_arguments(self, command)
+        try:
+            from azext_mlv2.manual._params import load_arguments as load_arguments_manual
+
+            load_arguments_manual(self, command)
+        except ImportError:
+            pass
+
+
+COMMAND_LOADER_CLS = AzureMachineLearningWorkspacesCommandsLoader

src/machinelearningservices/azext_mlv2/action.py

@@ -0,0 +1,18 @@
+# --------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for
+# license information.
+#
+# Code generated by Microsoft (R) AutoRest Code Generator.
+# Changes may cause incorrect behavior and will be lost if the code is
+# regenerated.
+# --------------------------------------------------------------------------
+# pylint: disable=wildcard-import
+# pylint: disable=unused-wildcard-import
+
+from .generated.action import *  # noqa: F403
+
+try:
+    from .manual.action import *  # noqa: F403
+except ImportError:
+    pass

src/machinelearningservices/azext_mlv2/azext_metadata.json

@@ -0,0 +1,3 @@
+{
+  "azext.minCliCoreVersion": "2.15.0"
+}

src/machinelearningservices/azext_mlv2/custom.py

@@ -0,0 +1,18 @@
+# --------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for
+# license information.
+#
+# Code generated by Microsoft (R) AutoRest Code Generator.
+# Changes may cause incorrect behavior and will be lost if the code is
+# regenerated.
+# --------------------------------------------------------------------------
+# pylint: disable=wildcard-import
+# pylint: disable=unused-wildcard-import
+
+from .generated.custom import *  # noqa: F403
+
+try:
+    from .manual.custom import *  # noqa: F403
+except ImportError:
+    pass

src/machinelearningservices/azext_mlv2/generated/__init__.py

@@ -0,0 +1,12 @@
+# coding=utf-8
+# --------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for
+# license information.
+#
+# Code generated by Microsoft (R) AutoRest Code Generator.
+# Changes may cause incorrect behavior and will be lost if the code is
+# regenerated.
+# --------------------------------------------------------------------------
+
+__path__ = __import__("pkgutil").extend_path(__path__, __name__)

src/machinelearningservices/azext_mlv2/generated/_client_factory.py

@@ -0,0 +1,20 @@
+# --------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for
+# license information.
+#
+# Code generated by Microsoft (R) AutoRest Code Generator.
+# Changes may cause incorrect behavior and will be lost if the code is
+# regenerated.
+# --------------------------------------------------------------------------
+
+
+def cf_ml_cl(cli_ctx, *_):
+    from azext_mlv2.manual.custom.utils import _get_cloud_information_from_cli
+
+    # the client used here doesn't matter since it isn't used by any command
+    from azure.ai.ml._restclient.v2022_02_01_preview import AzureMachineLearningWorkspaces
+    from azure.cli.core.commands.client_factory import get_mgmt_service_client
+
+    kwargs = _get_cloud_information_from_cli(cli_ctx=cli_ctx)
+    return get_mgmt_service_client(cli_ctx, AzureMachineLearningWorkspaces, **kwargs)

src/machinelearningservices/azext_mlv2/generated/_help.py

@@ -0,0 +1,12 @@
+# --------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for
+# license information.
+#
+# Code generated by Microsoft (R) AutoRest Code Generator.
+# Changes may cause incorrect behavior and will be lost if the code is
+# regenerated.
+# --------------------------------------------------------------------------
+# pylint: disable=too-many-lines
+
+from knack.help_files import helps

src/machinelearningservices/azext_mlv2/generated/_params.py

@@ -0,0 +1,14 @@
+# --------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for
+# license information.
+#
+# Code generated by Microsoft (R) AutoRest Code Generator.
+# Changes may cause incorrect behavior and will be lost if the code is
+# regenerated.
+# --------------------------------------------------------------------------
+# pylint: disable=unused-argument
+
+
+def load_arguments(self, _):
+    pass

src/machinelearningservices/azext_mlv2/generated/_validators.py

@@ -0,0 +1,9 @@
+# --------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for
+# license information.
+#
+# Code generated by Microsoft (R) AutoRest Code Generator.
+# Changes may cause incorrect behavior and will be lost if the code is
+# regenerated.
+# --------------------------------------------------------------------------

src/machinelearningservices/azext_mlv2/generated/action.py

@@ -0,0 +1,10 @@
+# --------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for
+# license information.
+#
+# Code generated by Microsoft (R) AutoRest Code Generator.
+# Changes may cause incorrect behavior and will be lost if the code is
+# regenerated.
+# --------------------------------------------------------------------------
+# pylint: disable=protected-access

src/machinelearningservices/azext_mlv2/generated/commands.py

@@ -0,0 +1,19 @@
+# --------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for
+# license information.
+#
+# Code generated by Microsoft (R) AutoRest Code Generator.
+# Changes may cause incorrect behavior and will be lost if the code is
+# regenerated.
+# --------------------------------------------------------------------------
+# pylint: disable=too-many-statements
+# pylint: disable=too-many-locals
+
+from azure.cli.core.commands import CliCommandType
+
+
+def load_command_table(self, _):
+
+    with self.command_group("ml", is_experimental=True):
+        pass

src/machinelearningservices/azext_mlv2/generated/custom.py

@@ -0,0 +1,10 @@
+# --------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for
+# license information.
+#
+# Code generated by Microsoft (R) AutoRest Code Generator.
+# Changes may cause incorrect behavior and will be lost if the code is
+# regenerated.
+# --------------------------------------------------------------------------
+# pylint: disable=too-many-lines

src/machinelearningservices/azext_mlv2/manual/.bumpversion.cfg

@@ -0,0 +1,3 @@
+[bumpversion]
+current_version = 0.0.139
+commit = True

src/machinelearningservices/azext_mlv2/manual/__init__.py

@@ -0,0 +1,12 @@
+# coding=utf-8
+# --------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for
+# license information.
+#
+# Code generated by Microsoft (R) AutoRest Code Generator.
+# Changes may cause incorrect behavior and will be lost if the code is
+# regenerated.
+# --------------------------------------------------------------------------
+
+__path__ = __import__('pkgutil').extend_path(__path__, __name__)