Addressing comments

ubuntu · Oct 30, 2023 · 30b66d9 · 30b66d9
1 parent 2236d73
commit 30b66d9
Showing 1 changed file with 5 additions and 8 deletions.
diff --git a/e2e/internal/az/image.go b/e2e/internal/az/image.go
@@ -1,6 +1,7 @@
 package az
 
 import (
+	"cmp"
 	"context"
 	"encoding/json"
 	"fmt"
@@ -72,10 +73,8 @@ func (images Images) LatestDaily() (Image, error) {
 
 	// Reverse sort images by version
 	slices.SortFunc(dailyImages, func(i, j Image) int {
-		if natural.Less(i.Version, j.Version) {
-			return -1
-		}
-		return 1
+		// Version format is: 23.04.20231029
+		return cmp.Compare(j.Version, i.Version)
 	})
 
 	return dailyImages[0], nil
@@ -97,10 +96,8 @@ func (images Images) LatestStable() (Image, error) {
 
 	// Reverse sort images by version
 	slices.SortFunc(stableImages, func(i, j Image) int {
-		if natural.Less(i.Version, j.Version) {
-			return -1
-		}
-		return 1
+		// Version format is: 23.04.20231029
+		return cmp.Compare(j.Version, i.Version)
 	})
 
 	return stableImages[0], nil