Update AzureMLKubernetes.py

To solve transient gate error
AzureArcForKubernetes · Jun 1, 2022 · dc14eb6 · dc14eb6
1 parent cc084eb
commit dc14eb6
Showing 1 changed file with 4 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
@@ -134,18 +134,18 @@ def Create(self, cmd, client, resource_group_name, cluster_name, name, cluster_t
                 cluster_resource_id, parent_api_version)
             cluster_location = resource.location.lower()
             try:
-                isSmallScale = False
+                isDevScale = False
                 if cluster_type.lower() == 'connectedclusters':
                     if resource.properties['totalNodeCount'] < 3:
-                        isSmallScale = True
+                        isDevScale = True
                 if cluster_type.lower() == 'managedclusters':
                     nodeCount = 0
                     for agent in resource.properties['agentPoolProfiles']:
                         nodeCount += agent['count']
                     if nodeCount < 3:
-                        isSmallScale = True
+                        isDevScale = True
 
-                if isSmallScale:
+                if isDevScale:
                     clusterPurpose = _get_value_from_config_protected_config(
                         self.clusterPurpose, configuration_settings, configuration_protected_settings)
                     if clusterPurpose is None: