Merge branch 'main' of https://github.com/moverseai/moai

moverseai · Jan 15, 2025 · c528592 · c528592
2 parents 1b05efe + ce780c2
commit c528592
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/moai/export/local/video2d.py b/moai/export/local/video2d.py
@@ -1,5 +1,6 @@
 import functools
 import logging
+import os
 import typing
 from collections import defaultdict
 
@@ -20,7 +21,7 @@
 def _create_video_writer(path: str, overwrite: bool, fps: float = 60.0):
     global _internal_video_id_
     vid = ffmpegio.open(
-        f"{path}/{_internal_video_id_}.mp4",
+        f"{path}/{_internal_video_id_}.mp4" if os.path.isdir(path) else path,
         "wv",
         fps,
         overwrite=overwrite,
@@ -57,7 +58,9 @@ def __init__(
         extension: str = "mp4",
         overwrite: bool = False,
     ):
-        self.path = ensure_path(log, "output folder", path)
+        self.path = (
+            ensure_path(log, "output folder", path) if os.path.isdir(path) else path
+        )
         self.format = ensure_choices(
             log, "output format", extension, MultiviewVideo2d.__FORMATS__
         )