Merge pull request #138 from agimus-project/ex-zip

unzip examples
agimus-project · Feb 21, 2024 · 67d74ce · 67d74ce
2 parents 06cfd35 + 75c136d
commit 67d74ce
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/happypose/toolbox/utils/download.py b/happypose/toolbox/utils/download.py
@@ -288,11 +288,9 @@ async def main():
             )
     if args.examples:
         for example in args.examples:
-            to_dl.append(
-                (
-                    f"examples/{example}",
-                    LOCAL_DATA_DIR / "examples",
-                )
+            to_dl.append((f"examples/{example}.zip", DOWNLOAD_DIR))
+            to_unzip.append(
+                (DOWNLOAD_DIR / f"{example}.zip", LOCAL_DATA_DIR / "examples")
             )
 
     # logger.info(f"{to_dl=}")