add customization for microsoft.azureml.kubernetes (#23)

* add customization for microsoft.azureml.kubernetes * Update release history Co-authored-by: Yue Yu <yuyu3@microsoft.com> Co-authored-by: jonathan-innis <jonathan.innis.ji@gmail.com>
AzureArcForKubernetes · Apr 23, 2021 · 5077192 · 5077192
1 parent 2971d69
commit 5077192
Show file tree

Hide file tree

Showing 4 changed files with 410 additions and 1 deletion.
diff --git a/src/k8s-extension/HISTORY.rst b/src/k8s-extension/HISTORY.rst
@@ -3,6 +3,11 @@
 Release History
 ===============
 
+0.3.0
+++++++++++++++++++
+
+* Release customization for microsoft.azureml.kubernetes
+
 0.2.1
 ++++++++++++++++++
 

diff --git a/src/k8s-extension/azext_k8s_extension/custom.py b/src/k8s-extension/azext_k8s_extension/custom.py
@@ -20,6 +20,7 @@
 from .partner_extensions.ContainerInsights import ContainerInsights
 from .partner_extensions.AzureDefender import AzureDefender
 from .partner_extensions.Cassandra import Cassandra
+from .partner_extensions.AzureMLKubernetes import AzureMLKubernetes
 from .partner_extensions.DefaultExtension import DefaultExtension
 from . import consts
 
@@ -33,6 +34,7 @@ def ExtensionFactory(extension_name):
     extension_map = {
         'microsoft.azuremonitor.containers': ContainerInsights,
         'microsoft.azuredefender.kubernetes': AzureDefender,
+        'microsoft.azureml.kubernetes': AzureMLKubernetes,
         'cassandradatacentersoperator': Cassandra,
     }