@@ -1393,31 +1393,32 @@ def set_up_blue_green_upgrade_settings(self, agentpool: AgentPool) -> AgentPool:
13931393 :return: the AgentPool object
13941394 """
13951395 self ._ensure_agentpool (agentpool )
1396-
1397- blue_green_upgrade_settings = self .models .AgentPoolBlueGreenUpgradeSettings ()
1398-
1396+
1397+ blue_green_upgrade_settings = self .models .AgentPoolBlueGreenUpgradeSettings () # pylint: disable=no-member
1398+
13991399 # Set each field if provided in context
14001400 drain_batch_size = self .context .get_drain_batch_size ()
14011401 if drain_batch_size is not None :
14021402 blue_green_upgrade_settings .drain_batch_size = drain_batch_size
1403-
1403+
14041404 drain_timeout_bg = self .context .get_drain_timeout_bg ()
14051405 if drain_timeout_bg is not None :
14061406 blue_green_upgrade_settings .drain_timeout_in_minutes = drain_timeout_bg
1407-
1407+
14081408 batch_soak_duration = self .context .get_batch_soak_duration ()
14091409 if batch_soak_duration is not None :
14101410 blue_green_upgrade_settings .batch_soak_duration_in_minutes = batch_soak_duration
1411-
1411+
14121412 final_soak_duration = self .context .get_final_soak_duration ()
14131413 if final_soak_duration is not None :
14141414 blue_green_upgrade_settings .final_soak_duration_in_minutes = final_soak_duration
1415-
1415+
14161416 # Set the blue-green upgrade settings as a separate property on agentpool
14171417 agentpool .upgrade_settings_blue_green = blue_green_upgrade_settings
14181418
14191419 return agentpool
14201420
1421+
14211422class AKSPreviewAgentPoolUpdateDecorator (AKSAgentPoolUpdateDecorator ):
14221423 def __init__ (
14231424 self ,
@@ -1724,7 +1725,7 @@ def update_upgrade_settings(self, agentpool: AgentPool) -> AgentPool:
17241725 upgrade_settings .max_unavailable = max_unavailable
17251726
17261727 return agentpool
1727-
1728+
17281729 def update_blue_green_upgrade_settings (self , agentpool : AgentPool ) -> AgentPool :
17291730 """Update blue-green upgrade settings for the AgentPool object.
17301731
@@ -1742,20 +1743,20 @@ def update_blue_green_upgrade_settings(self, agentpool: AgentPool) -> AgentPool:
17421743 # Initialize blue-green upgrade settings with existing values as defaults
17431744 upgrade_settings_blue_green = agentpool .upgrade_settings_blue_green
17441745 if upgrade_settings_blue_green is None :
1745- upgrade_settings_blue_green = self .models .AgentPoolBlueGreenUpgradeSettings ()
1746-
1746+ upgrade_settings_blue_green = self .models .AgentPoolBlueGreenUpgradeSettings () # pylint: disable=no-member
1747+
17471748 if drain_batch_size is not None :
17481749 upgrade_settings_blue_green .drain_batch_size = drain_batch_size
1749-
1750+
17501751 if drain_timeout_bg is not None :
17511752 upgrade_settings_blue_green .drain_timeout_in_minutes = drain_timeout_bg
1752-
1753+
17531754 if batch_soak_duration is not None :
17541755 upgrade_settings_blue_green .batch_soak_duration_in_minutes = batch_soak_duration
17551756
17561757 if final_soak_duration is not None :
1757- upgrade_settings_blue_green .final_soak_duration_in_minutes = final_soak_duration
1758-
1758+ upgrade_settings_blue_green .final_soak_duration_in_minutes = final_soak_duration
1759+
17591760 # Set the blue-green upgrade settings as a separate property on agentpool
17601761 agentpool .upgrade_settings_blue_green = upgrade_settings_blue_green
17611762
0 commit comments