From 1c55e2fbaa98b2497c4d9d36bb1fbd32ec194238 Mon Sep 17 00:00:00 2001
From: imagineer99 <samleejackson0@gmail.com>
Date: Thu, 26 Feb 2026 03:57:55 +0000
Subject: [PATCH] fix: remove dataset metadata badges from HF dataset dropdowns

---
 .../components/steps/dataset-step.tsx         | 21 ++-----------------
 .../studio/sections/dataset-section.tsx       | 21 +------------------
 2 files changed, 3 insertions(+), 39 deletions(-)

diff --git a/studio/frontend/src/features/onboarding/components/steps/dataset-step.tsx b/studio/frontend/src/features/onboarding/components/steps/dataset-step.tsx
index 1ad4fad8f0..99434f1285 100644
--- a/studio/frontend/src/features/onboarding/components/steps/dataset-step.tsx
+++ b/studio/frontend/src/features/onboarding/components/steps/dataset-step.tsx
@@ -38,7 +38,7 @@ import {
   useHfTokenValidation,
   useInfiniteScroll,
 } from "@/hooks";
-import { cn, formatCompact } from "@/lib/utils";
+import { cn } from "@/lib/utils";
 import {
   HfDatasetSubsetSplitSelectors,
   useTrainingConfigStore,
@@ -251,16 +251,8 @@ export function DatasetStep() {
                   >
                     <ComboboxList className="p-1 !max-h-none !overflow-visible">
                       {(id: string) => {
-                        const r = hfResults.find((r) => r.id === id);
-                        const detail = r?.totalExamples
-                          ? `${formatCompact(r.totalExamples)} rows`
-                          : (r?.sizeCategory ?? null);
                         return (
-                          <ComboboxItem
-                            key={id}
-                            value={id}
-                            className="justify-between"
-                          >
+                          <ComboboxItem key={id} value={id}>
                             <Tooltip>
                               <TooltipTrigger asChild={true}>
                                 <span className="min-w-0 flex-1 truncate">
@@ -274,15 +266,6 @@ export function DatasetStep() {
                                 {id}
                               </TooltipContent>
                             </Tooltip>
-                            {detail ? (
-                              <span className="text-[10px] text-muted-foreground shrink-0">
-                                {detail}
-                              </span>
-                            ) : r?.downloads != null ? (
-                              <span className="text-[10px] text-muted-foreground shrink-0">
-                                ↓{formatCompact(r.downloads)}
-                              </span>
-                            ) : null}
                           </ComboboxItem>
                         );
                       }}
diff --git a/studio/frontend/src/features/studio/sections/dataset-section.tsx b/studio/frontend/src/features/studio/sections/dataset-section.tsx
index 6f6a6a401e..cf0fb7d90f 100644
--- a/studio/frontend/src/features/studio/sections/dataset-section.tsx
+++ b/studio/frontend/src/features/studio/sections/dataset-section.tsx
@@ -33,7 +33,6 @@ import {
   useHfTokenValidation,
   useInfiniteScroll,
 } from "@/hooks";
-import { formatCompact } from "@/lib/utils";
 import {
   HfDatasetSubsetSplitSelectors,
   useDatasetPreviewDialogStore,
@@ -221,21 +220,8 @@ export function DatasetSection() {
                   >
                     <ComboboxList className="p-1 !max-h-none !overflow-visible">
                       {(id: string) => {
-                        const r = hfResults.find((ds) => ds.id === id);
-                        let detail: string | null = null;
-                        if (r?.totalExamples) {
-                          detail = `${formatCompact(r.totalExamples)} rows`;
-                        } else if (r?.sizeCategory) {
-                          detail = r.sizeCategory;
-                        } else if (r?.downloads != null) {
-                          detail = `↓${formatCompact(r.downloads)}`;
-                        }
                         return (
-                          <ComboboxItem
-                            key={id}
-                            value={id}
-                            className="justify-between"
-                          >
+                          <ComboboxItem key={id} value={id}>
                             <Tooltip>
                               <TooltipTrigger asChild={true}>
                                 <span className="min-w-0 flex-1 truncate">
@@ -249,11 +235,6 @@ export function DatasetSection() {
                                 {id}
                               </TooltipContent>
                             </Tooltip>
-                            {detail && (
-                              <span className="text-[10px] text-muted-foreground shrink-0">
-                                {detail}
-                              </span>
-                            )}
                           </ComboboxItem>
                         );
                       }}