Make VmSize validation logic clearer

Azure · Jan 31, 2025 · 434650e · 434650e
1 parent a94fa96
commit 434650e
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/pkg/api/validate/vm.go b/pkg/api/validate/vm.go
@@ -247,16 +247,12 @@ func VMSizeIsValid(vmSize api.VMSize, requiredD2sV3Workers, isMaster bool) bool
 		return supportedAsMaster
 	}
 
-	if requiredD2sV3Workers && vmSize != api.VMSizeStandardD2sV3 {
-		return false
+	if requiredD2sV3Workers {
+		return vmSize == api.VMSizeStandardD2sV3
 	}
 
 	_, supportedAsWorker := SupportedVMSizesByRole(VMRoleWorker)[vmSize]
-	if supportedAsWorker || (requiredD2sV3Workers && vmSize == api.VMSizeStandardD2sV3) {
-		return true
-	}
-
-	return false
+	return supportedAsWorker
 }
 
 func VMSizeFromName(vmSize api.VMSize) (api.VMSizeStruct, bool) {