Rename inferenceLoadBalancerHA (#123)

* Rename inferenceLoadBalancerHA Rename inferenceLoadBalancerHA to inferenceRouterHA and unify related logic * Update AzureMLKubernetes.py * Remove unused comments to trigger pipeline * Update AzureMLKubernetes.py
AzureArcForKubernetes · May 12, 2022 · c8afe1a · c8afe1a
1 parent 594c2e9
commit c8afe1a
Showing 1 changed file with 11 additions and 12 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
@@ -83,7 +83,7 @@ def __init__(self):
 
         self.inferenceRouterServiceType = 'inferenceRouterServiceType'
         self.internalLoadBalancerProvider = 'internalLoadBalancerProvider'
-        self.inferenceLoadBalancerHA = 'inferenceLoadBalancerHA'
+        self.inferenceRouterHA = 'inferenceRouterHA'
 
         # constants for existing AKS to AMLARC migration
         self.IS_AKS_MIGRATION = 'isAKSMigration'
@@ -322,14 +322,13 @@ def Update(self, cmd, resource_group_name, cluster_name, auto_upgrade_minor_vers
 
     def __normalize_config(self, configuration_settings, configuration_protected_settings):
         # inference
-        isTestCluster = _get_value_from_config_protected_config(
-            self.inferenceLoadBalancerHA, configuration_settings, configuration_protected_settings)
-        if isTestCluster is not None:
-            isTestCluster = str(isTestCluster).lower() == 'false'
-            if isTestCluster:
-                configuration_settings['clusterPurpose'] = 'DevTest'
-            else:
-                configuration_settings['clusterPurpose'] = 'FastProd'
+        inferenceRouterHA = _get_value_from_config_protected_config(
+            self.inferenceRouterHA, configuration_settings, configuration_protected_settings)
+        isTestCluster = True if inferenceRouterHA is not None and str(inferenceRouterHA).lower() == 'false' else False
+        if isTestCluster:
+            configuration_settings['clusterPurpose'] = 'DevTest'
+        else:
+            configuration_settings['clusterPurpose'] = 'FastProd'
 
         inferenceRouterServiceType = _get_value_from_config_protected_config(
             self.inferenceRouterServiceType, configuration_settings, configuration_protected_settings)
@@ -386,9 +385,9 @@ def __validate_config(self, configuration_settings, configuration_protected_sett
         configuration_protected_settings.pop(self.ENABLE_INFERENCE, None)
 
     def __validate_scoring_fe_settings(self, configuration_settings, configuration_protected_settings, release_namespace):
-        isTestCluster = _get_value_from_config_protected_config(
-            self.inferenceLoadBalancerHA, configuration_settings, configuration_protected_settings)
-        isTestCluster = str(isTestCluster).lower() == 'false'
+        inferenceRouterHA = _get_value_from_config_protected_config(
+            self.inferenceRouterHA, configuration_settings, configuration_protected_settings)
+        isTestCluster = True if inferenceRouterHA is not None and str(inferenceRouterHA).lower() == 'false' else False
         if isTestCluster:
             configuration_settings['clusterPurpose'] = 'DevTest'
         else: