visualize_dataset / src /lib /so101-robot.ts
mishig's picture
mishig HF Staff
Sync from GitHub via hub-sync
ffcdaff verified
export function isSO101Robot(robotType: string | null): boolean {
if (!robotType) return false;
const lower = robotType.toLowerCase();
return (
lower.includes("so100") ||
lower.includes("so101") ||
lower === "so_follower"
);
}
export function isOpenArmRobot(robotType: string | null): boolean {
if (!robotType) return false;
return robotType.toLowerCase().includes("openarm");
}
export function isG1Robot(robotType: string | null): boolean {
if (!robotType) return false;
const lower = robotType.toLowerCase();
return lower.includes("g1") || lower.includes("unitree");
}
export function hasURDFSupport(robotType: string | null): boolean {
return (
isSO101Robot(robotType) || isOpenArmRobot(robotType) || isG1Robot(robotType)
);
}