From c2fe0f8fe6b77ae167e99653cd4d43922cdcfad1 Mon Sep 17 00:00:00 2001 From: Yue Yu Date: Tue, 9 Nov 2021 17:12:55 +0800 Subject: [PATCH] feIsNodePort, feIsInternalLoadBalancer --- .../partner_extensions/AzureMLKubernetes.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/src/k8s-extension/azext_k8s_extension/partner_extensions/AzureMLKubernetes.py b/src/k8s-extension/azext_k8s_extension/partner_extensions/AzureMLKubernetes.py index fc601c46327..706973d9a13 100644 --- a/src/k8s-extension/azext_k8s_extension/partner_extensions/AzureMLKubernetes.py +++ b/src/k8s-extension/azext_k8s_extension/partner_extensions/AzureMLKubernetes.py @@ -226,6 +226,20 @@ def __normalize_config(self, configuration_settings, configuration_protected_set else: configuration_settings['clusterPurpose'] = 'FastProd' + feIsNodePort = _get_value_from_config_protected_config( + self.privateEndpointNodeport, configuration_settings, configuration_protected_settings) + if feIsNodePort is not None: + feIsNodePort = str(feIsNodePort).lower() == 'true' + configuration_settings['scoringFe.serviceType.nodePort'] = feIsNodePort + + feIsInternalLoadBalancer = _get_value_from_config_protected_config( + self.privateEndpointILB, configuration_settings, configuration_protected_settings) + if feIsInternalLoadBalancer is not None: + feIsInternalLoadBalancer = str(feIsInternalLoadBalancer).lower() == 'true' + configuration_settings['scoringFe.serviceType.internalLoadBalancer'] = feIsInternalLoadBalancer + logger.warning( + 'Internal load balancer only supported on AKS and AKS Engine Clusters.') + def __validate_config(self, configuration_settings, configuration_protected_settings): # perform basic validation of the input config config_keys = configuration_settings.keys()