From 208a88bfd57a1ceb1cf6e0b44b4155bdc5bceed5 Mon Sep 17 00:00:00 2001
From: Nicolas Mowen <nickmowen213@gmail.com>
Date: Sat, 28 Jun 2025 13:45:11 -0600
Subject: [PATCH] Sort class images

---
 web/src/views/classification/ModelTrainingView.tsx | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/web/src/views/classification/ModelTrainingView.tsx b/web/src/views/classification/ModelTrainingView.tsx
index 14de1a118..145004ec3 100644
--- a/web/src/views/classification/ModelTrainingView.tsx
+++ b/web/src/views/classification/ModelTrainingView.tsx
@@ -577,9 +577,14 @@ function DatasetGrid({
 }: DatasetGridProps) {
   const { t } = useTranslation(["views/classificationModel"]);
 
+  const classData = useMemo(
+    () => images.sort((a, b) => a.localeCompare(b)),
+    [images],
+  );
+
   return (
     <div className="flex flex-wrap gap-2 overflow-y-auto p-2">
-      {images.map((image) => (
+      {classData.map((image) => (
         <div
           className={cn(
             "flex w-60 cursor-pointer flex-col gap-2 rounded-lg bg-card outline outline-[3px]",