feat: added datasets of size 5M and 10M to pretraining size category

2026-02-26 06:32:45 +00:00 · 2026-02-26 06:32:45 +00:00 · 852dff564e
commit 852dff564e
parent 6e535ed0eb
1 changed files with 2 additions and 0 deletions
--- a/studio/frontend/src/hooks/use-hf-dataset-search.ts
+++ b/studio/frontend/src/hooks/use-hf-dataset-search.ts
@ -141,6 +141,8 @@ const PRETRAINING_PLAIN_TAGS = new Set(["pretraining", "pre-training"]);
 const OCR_PLAIN_TAGS = new Set(["ocr", "document-ocr"]);

 const PRETRAINING_SIZE_CATEGORIES = new Set([
+  "5M<n<10M",
+  "10M<n<100M",
  "100M<n<1B",
  "1B<n<10B",
  "10B<n<100B",