admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-21 17:01:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix default version to v2.1 for vizualizer

Browse Source
This commit is contained in:
Michel Aractingi
2025-07-22 10:19:38 +02:00
parent 5e86d6228c
commit 2efc74012d
2 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
lerobot/html_dataset_visualizer/src/app/[org]/[dataset]/[episode]/fetch-data.ts
+5 -3
View File
@@ -10,6 +10,8 @@ import { pick } from "@/utils/pick";
const DATASET_URL = const DATASET_URL =
process.env.DATASET_URL || "https://huggingface.co/datasets"; process.env.DATASET_URL || "https://huggingface.co/datasets";
const DEFAULT_REVISION = "v2.1";
const SERIES_NAME_DELIMITER = " | "; const SERIES_NAME_DELIMITER = " | ";
export async function getEpisodeData( export async function getEpisodeData(
@@ -20,7 +22,7 @@ export async function getEpisodeData(
const repoId = `${org}/${dataset}`; const repoId = `${org}/${dataset}`;
try { try {
const episode_chunk = Math.floor(0 / 1000); const episode_chunk = Math.floor(0 / 1000);
const jsonUrl = `${DATASET_URL}/${repoId}/resolve/main/meta/info.json`; const jsonUrl = `${DATASET_URL}/${repoId}/resolve/${DEFAULT_REVISION}/meta/info.json`;
const info = await fetchJson<DatasetMetadata>(jsonUrl); const info = await fetchJson<DatasetMetadata>(jsonUrl);
@@ -56,7 +58,7 @@ export async function getEpisodeData(
}); });
return { return {
filename: key, filename: key,
url: `${DATASET_URL}/${repoId}/resolve/main/` + videoPath, url: `${DATASET_URL}/${repoId}/resolve/${DEFAULT_REVISION}/` + videoPath,
}; };
}); });
@@ -103,7 +105,7 @@ export async function getEpisodeData(
}); });
const parquetUrl = const parquetUrl =
`${DATASET_URL}/${repoId}/resolve/main/` + `${DATASET_URL}/${repoId}/resolve/${DEFAULT_REVISION}/` +
formatStringWithVars(info.data_path, { formatStringWithVars(info.data_path, {
episode_chunk: episode_chunk.toString().padStart(3, "0"), episode_chunk: episode_chunk.toString().padStart(3, "0"),
episode_index: episodeId.toString().padStart(6, "0"), episode_index: episodeId.toString().padStart(6, "0"),
lerobot/html_dataset_visualizer/src/app/explore/page.tsx
+5 -2
View File
@@ -38,6 +38,9 @@ export default async function ExplorePage({
return <div className="p-8 text-red-600">Failed to load datasets.</div>; return <div className="p-8 text-red-600">Failed to load datasets.</div>;
} }
// Default to v2.1 revision for dataset loading
const DEFAULT_REVISION = "v2.1";
// Fetch episode 0 data for each dataset // Fetch episode 0 data for each dataset
const datasetWithVideos = ( const datasetWithVideos = (
await Promise.all( await Promise.all(
@@ -45,7 +48,7 @@ export default async function ExplorePage({
try { try {
const [org, dataset] = ds.id.split("/"); const [org, dataset] = ds.id.split("/");
const repoId = `${org}/${dataset}`; const repoId = `${org}/${dataset}`;
const jsonUrl = `https://huggingface.co/datasets/${repoId}/resolve/main/meta/info.json`; const jsonUrl = `https://huggingface.co/datasets/${repoId}/resolve/${DEFAULT_REVISION}/meta/info.json`;
const info = await fetchJson<DatasetMetadata>(jsonUrl); const info = await fetchJson<DatasetMetadata>(jsonUrl);
const videoEntry = Object.entries(info.features).find( const videoEntry = Object.entries(info.features).find(
([key, value]) => value.dtype === "video", ([key, value]) => value.dtype === "video",
@@ -59,7 +62,7 @@ export default async function ExplorePage({
episode_index: "0".padStart(6, "0"), episode_index: "0".padStart(6, "0"),
}); });
const url = const url =
`https://huggingface.co/datasets/${repoId}/resolve/main/` + `https://huggingface.co/datasets/${repoId}/resolve/${DEFAULT_REVISION}/` +
videoPath; videoPath;
// Check if videoUrl exists (status 200) // Check if videoUrl exists (status 200)
try { try {