- {hasMultipleConfigs && (
-
-
- Subset (Config)
-
-
-
-
-
- This dataset has multiple subsets (configurations).
- Select which one to use for training.
-
-
-
-
-
- )}
-
- {hasMultipleSplits && (
-
-
- Split
-
-
-
-
-
- Select which split of the dataset to use for training.
-
-
-
-
-
- )}
-
diff --git a/studio/frontend/src/features/training/components/hf-dataset-config-split-selectors.tsx b/studio/frontend/src/features/training/components/hf-dataset-config-split-selectors.tsx
new file mode 100644
index 0000000000..4342ac44f0
--- /dev/null
+++ b/studio/frontend/src/features/training/components/hf-dataset-config-split-selectors.tsx
@@ -0,0 +1,271 @@
+import {
+ Select,
+ SelectContent,
+ SelectItem,
+ SelectTrigger,
+ SelectValue,
+} from "@/components/ui/select";
+import { Spinner } from "@/components/ui/spinner";
+import {
+ Tooltip,
+ TooltipContent,
+ TooltipTrigger,
+} from "@/components/ui/tooltip";
+import {
+ Field,
+ FieldLabel,
+} from "@/components/ui/field";
+import { useHfDatasetSplits } from "@/hooks";
+import { InformationCircleIcon } from "@hugeicons/core-free-icons";
+import { HugeiconsIcon } from "@hugeicons/react";
+import { useEffect } from "react";
+
+type Props = {
+ variant: "wizard" | "studio";
+ enabled: boolean;
+ datasetName: string | null;
+ accessToken?: string;
+ datasetConfig: string | null;
+ setDatasetConfig: (v: string | null) => void;
+ datasetSplit: string | null;
+ setDatasetSplit: (v: string | null) => void;
+};
+
+export function HfDatasetConfigSplitSelectors({
+ variant,
+ enabled,
+ datasetName,
+ accessToken,
+ datasetConfig,
+ setDatasetConfig,
+ datasetSplit,
+ setDatasetSplit,
+}: Props) {
+ const {
+ configs: hfConfigs,
+ splits: hfSplits,
+ hasMultipleConfigs,
+ hasMultipleSplits,
+ isLoading,
+ error,
+ } = useHfDatasetSplits(enabled ? datasetName : null, datasetConfig, {
+ accessToken,
+ });
+
+ useEffect(() => {
+ if (hfConfigs.length === 1 && datasetConfig !== hfConfigs[0]) {
+ setDatasetConfig(hfConfigs[0]);
+ }
+ }, [hfConfigs, datasetConfig, setDatasetConfig]);
+
+ useEffect(() => {
+ if (hfSplits.length === 0) return;
+ if (hasMultipleConfigs && !datasetConfig) return;
+ if (hfSplits.length === 1 && datasetSplit !== hfSplits[0]) {
+ setDatasetSplit(hfSplits[0]);
+ } else if (!datasetSplit && hfSplits.includes("train")) {
+ setDatasetSplit("train");
+ } else if (!datasetSplit) {
+ setDatasetSplit(hfSplits[0]);
+ }
+ }, [
+ hfSplits,
+ hasMultipleConfigs,
+ datasetConfig,
+ datasetSplit,
+ setDatasetSplit,
+ ]);
+
+ if (!enabled || !datasetName) return null;
+
+ return (
+ <>
+ {isLoading && (
+
+
+ Loading dataset configs and splits...
+
+ )}
+
+ {error && (
+
+ Could not fetch dataset splits: {error}
+
+ )}
+
+ {!isLoading && !error && hasMultipleConfigs && (
+ <>
+ {variant === "wizard" ? (
+
+
+ Subset (Config)
+
+
+
+
+
+ This dataset has multiple subsets (configurations). Select
+ which one to use for training.
+
+
+
+
+
+ ) : (
+
+
+ Subset (Config)
+
+
+
+
+
+ This dataset has multiple subsets (configurations). Select
+ which one to use for training.
+
+
+
+
+
+ )}
+
+ )}
+
+ {!isLoading && !error && hasMultipleSplits && (
+ <>
+ {variant === "wizard" ? (
+
+
+ Split
+
+
+
+
+
+ Select which split of the dataset to use for training.
+
+
+
+
+
+ ) : (
+
+
+ Split
+
+
+
+
+
+ Select which split of the dataset to use for training.
+
+
+
+
+
+ )}
+
+ )}
+
+ );
+}
diff --git a/studio/frontend/src/features/training/index.ts b/studio/frontend/src/features/training/index.ts
index 48b9079fd3..d32332a19b 100644
--- a/studio/frontend/src/features/training/index.ts
+++ b/studio/frontend/src/features/training/index.ts
@@ -5,4 +5,5 @@ export {
} from "./stores/training-runtime-store";
export { useTrainingActions } from "./hooks/use-training-actions";
export { useTrainingRuntimeLifecycle } from "./hooks/use-training-runtime-lifecycle";
+export { HfDatasetConfigSplitSelectors } from "./components/hf-dataset-config-split-selectors";
export type { TrainingPhase } from "./types/runtime";
diff --git a/studio/frontend/src/hooks/use-hf-dataset-splits.ts b/studio/frontend/src/hooks/use-hf-dataset-splits.ts
index b09e1e3d60..ededfacbd6 100644
--- a/studio/frontend/src/hooks/use-hf-dataset-splits.ts
+++ b/studio/frontend/src/hooks/use-hf-dataset-splits.ts
@@ -117,10 +117,14 @@ export function useHfDatasetSplits(
// Derive unique configs
const configs = Array.from(new Set(entries.map((e) => e.config)));
- // Derive splits for the selected config (or all splits if no config selected)
- const filteredEntries = selectedConfig
- ? entries.filter((e) => e.config === selectedConfig)
- : entries;
+ // Derive splits for the active config.
+ // If dataset has >1 config and none is selected yet, return no splits so UI
+ // doesn't auto-pick/show a split before config is chosen.
+ const activeConfig =
+ selectedConfig ?? (configs.length === 1 ? configs[0] : null);
+ const filteredEntries = activeConfig
+ ? entries.filter((e) => e.config === activeConfig)
+ : [];
const splits = Array.from(new Set(filteredEntries.map((e) => e.split)));
return {
@@ -128,7 +132,7 @@ export function useHfDatasetSplits(
splits,
entries,
hasMultipleConfigs: configs.length > 1,
- hasMultipleSplits: splits.length > 1,
+ hasMultipleSplits: activeConfig ? splits.length > 1 : false,
isLoading,
error,
};