fix install issue on aks (#119)

* fix install issue on aks * Removing checking distribution Co-authored-by: youhuatu <youhuatu@outlook.com>
AzureArcForKubernetes · Apr 15, 2022 · 37fc852 · 37fc852
1 parent 7e5d012
commit 37fc852
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/src/k8s-extension/azext_k8s_extension/partner_extensions/AzureMLKubernetes.py b/src/k8s-extension/azext_k8s_extension/partner_extensions/AzureMLKubernetes.py
@@ -126,10 +126,18 @@ def Create(self, cmd, client, resource_group_name, cluster_name, name, cluster_t
             resource = resources.get_by_id(
                 cluster_resource_id, parent_api_version)
             cluster_location = resource.location.lower()
-            if resource.properties['totalNodeCount'] == 1 or resource.properties['totalNodeCount'] == 2:
-                configuration_settings['clusterPurpose'] = 'DevTest'
-            if resource.properties['distribution'].lower() == "openshift":
-                configuration_settings[self.OPEN_SHIFT] = "true"
+            try:
+                if cluster_type.lower() == 'connectedclusters':
+                    if resource.properties['totalNodeCount'] < 3:
+                        configuration_settings['clusterPurpose'] = 'DevTest'
+                if cluster_type.lower() == 'managedclusters':
+                    nodeCount = 0
+                    for agent in resource.properties['agentPoolProfiles']:
+                        nodeCount += agent['count']
+                    if nodeCount < 3:
+                        configuration_settings['clusterPurpose'] = 'DevTest'
+            except:
+                pass
         except CloudError as ex:
             raise ex