From 4d2275635a6220e875bf56b213cd23ca3ba69add Mon Sep 17 00:00:00 2001
From: hrntsm <contact@hrntsm.com>
Date: Sat, 12 Aug 2023 13:54:34 +0900
Subject: [PATCH] Add model viewer target ext array

---
 optuna_dashboard/ts/components/TrialList.tsx | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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<HTMLInputElement>(null)
   const handleClick: MouseEventHandler = () => {
@@ -445,8 +446,7 @@ const TrialArtifact: FC<{ trial: Trial }> = ({ trial }) => {
               </Card>
             )
           } else if (
-            a.filename.endsWith(".stl") ||
-            a.filename.endsWith(".3dm")
+            modelViewerTargetExt.includes(a.filename.split(".").pop() || "")
           ) {
             return (
               <Card