From ddb66703a19cef227a97938c9a28dde4e3646da9 Mon Sep 17 00:00:00 2001
From: dongchangxi <458593490@qq.com>
Date: Mon, 17 Mar 2025 10:01:03 +0800
Subject: [PATCH] 2

---
 tools/cal_weight.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/tools/cal_weight.py b/tools/cal_weight.py
index 9227fec..121fbbd 100644
--- a/tools/cal_weight.py
+++ b/tools/cal_weight.py
@@ -100,17 +100,17 @@ def down_obj_from_oss(workdir, pid, action):
                 isMtlFile = True
                 
         
-        if localfile.endswith('.obj'):
+        if "obj" in localfile:
             if isObjFile:
                 print(f'目录{dirPath}已存在obj文件，跳过')
                 continue
         
-        if localfile.endswith('.jpg'):  
+        if "jpg" in localfile:  
             if isJpgFile:
                 print(f'目录{dirPath}已存在jpg文件，跳过')
                 continue
         
-        if localfile.endswith('.mtl'):
+        if "mtl" in localfile:
             if isMtlFile:
                 print(f'目录{dirPath}已存在mtl文件，跳过')
                 continue