bug fixes (#25671)

Azure · Jul 29, 2024 · 3ca5ebc · 3ca5ebc
1 parent 475411b
commit 3ca5ebc
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 7 deletions.
diff --git a/src/AksArc/AksArc.Autorest/custom/Invoke-AzAksArcClusterUpgrade.ps1 b/src/AksArc/AksArc.Autorest/custom/Invoke-AzAksArcClusterUpgrade.ps1
@@ -121,11 +121,9 @@ function Invoke-AzAksArcClusterUpgrade {
         if ($PSBoundParameters.ContainsKey("KubernetesVersion"))
         {   
             $Upgrades = Get-AzAksArcClusterUpgrade -ClusterName $ClusterName -ResourceGroupName $ResourceGroupName -SubscriptionId $SubscriptionId
-            if ($upgrades.ControlPlaneProfileUpgrade.KubernetesVersion -contains $KubernetesVersion) {
-                continue
-            } else {
+            if (!($upgrades.ControlPlaneProfileUpgrade.KubernetesVersion -contains $KubernetesVersion)) {
                 throw "Kubernetes Version $KubernetesVersion is not a valid upgradable version."
-            }
+            } 
         } else {
             $Upgrades = Get-AzAksArcClusterUpgrade -ClusterName $ClusterName -ResourceGroupName $ResourceGroupName -SubscriptionId $SubscriptionId
             $UpgradeListLength = $upgrades.ControlPlaneProfileUpgrade.KubernetesVersion.Length
@@ -137,8 +135,7 @@ function Invoke-AzAksArcClusterUpgrade {
 
             $null = $PSBoundParameters.Add("KubernetesVersion", $LatestUpgrade)
         }
-
+        
         Az.AksArc.internal\Update-AzAksArcCluster @PSBoundParameters
     }
-    }
-
+    }
diff --git a/src/AksArc/AksArc.Autorest/custom/New-AzAksArcCluster.ps1 b/src/AksArc/AksArc.Autorest/custom/New-AzAksArcCluster.ps1
@@ -475,6 +475,8 @@ function New-AzAksArcCluster {
             $null = $PSBoundParameters.Add("LicenseProfileAzureHybridBenefit", $false)
         }
         $null = $PSBoundParameters.Remove("EnableAzureHybridBenefit")
+        $null = $PSBoundParameters.Remove("NodeTaint")
+        $null = $PSBoundParameters.Remove("NodeLabel")
         Az.AksArc.internal\New-AzAksArcCluster @PSBoundParameters
     }
     }