modify condition in download_weights for testing

developer0hye · Oct 15, 2023 · 75e8ae0 · 75e8ae0
1 parent fb4c4d1
commit 75e8ae0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onepose/utils/__init__.py b/onepose/utils/__init__.py
@@ -19,7 +19,7 @@ def update_to(self, b=1, bsize=1, tsize=None):
 
 def download_weights(url: str, filename: str, hash: str) -> None:
     if os.path.exists(filename):
-        if calc_file_hash(filename) == hash:
+        if calc_file_hash(filename) == hash or hash == 'test':
             print(f'{filename} is already downloaded.')
             return True
         os.remove(filename)