diff --git a/optuna_dashboard/ts/components/TrialList.tsx b/optuna_dashboard/ts/components/TrialList.tsx index 9a5bbae9..8c70d554 100644 --- a/optuna_dashboard/ts/components/TrialList.tsx +++ b/optuna_dashboard/ts/components/TrialList.tsx @@ -336,6 +336,7 @@ const TrialArtifact: FC<{ trial: Trial }> = ({ trial }) => { const width = "200px" const height = "150px" + const modelViewerTargetExt = ["stl", "3dm"] const inputRef = useRef(null) const handleClick: MouseEventHandler = () => { @@ -445,8 +446,7 @@ const TrialArtifact: FC<{ trial: Trial }> = ({ trial }) => { ) } else if ( - a.filename.endsWith(".stl") || - a.filename.endsWith(".3dm") + modelViewerTargetExt.includes(a.filename.split(".").pop() || "") ) { return (