Merge AzureMLKubernetes Changes into private preview branch

src/k8s-extension/azext_k8s_extension/_consts_private.py

@@ -5,4 +5,4 @@
 # --------------------------------------------------------------------------------------------
 
 EXTENSION_NAME = 'k8s-extension-private'
-VERSION = "0.1PP.14"
+VERSION = "0.2.1"

src/k8s-extension/azext_k8s_extension/custom.py

@@ -21,6 +21,7 @@
 from azext_k8s_extension.partner_extensions.AzureDefender import AzureDefender
 from azext_k8s_extension.partner_extensions.Cassandra import Cassandra
 from azext_k8s_extension.partner_extensions.OpenServiceMesh import OpenServiceMesh
+from azext_k8s_extension.partner_extensions.AzureMLKubernetes import AzureMLKubernetes
 from azext_k8s_extension.partner_extensions.DefaultExtension import DefaultExtension
 import azext_k8s_extension._consts as consts
 
@@ -35,6 +36,7 @@ def ExtensionFactory(extension_name):
         'microsoft.azuremonitor.containers': ContainerInsights,
         'microsoft.azuredefender.kubernetes': AzureDefender,
         'microsoft.openservicemesh': OpenServiceMesh,
+        'microsoft.azureml.kubernetes': AzureMLKubernetes,
         'cassandradatacentersoperator': Cassandra
     }