make sure vid is ok too

nelsonjchen · Dec 22, 2024 · ab53509 · ab53509
1 parent 20bc603
commit ab53509
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/Makefile b/Makefile
@@ -87,6 +87,10 @@ predict-bug-all-number:
 	./cog/generate.sh
 	cog predict -i route="https://connect.comma.ai/fe18f736cb0d7813/00000497--5809888120/1611/1635" -i renderType=ui
 
+predict-bug-all-number-360:
+	./cog/generate.sh
+	cog predict -i route="https://connect.comma.ai/fe18f736cb0d7813/00000497--5809888120/1611/1635" -i renderType=360
+
 
 # Push using modified cog
 push:

diff --git a/ffmpeg_clip.py b/ffmpeg_clip.py
@@ -59,7 +59,7 @@ def make_ffmpeg_clip(
         raise ValueError(f"Invalid choice: {render_type}")
     if not os.path.exists(data_dir):
         raise ValueError(f"Invalid data_dir: {data_dir}")
-    route = re.sub(r"--\d+$", "", route_or_segment)
+    route = re.sub(r"--\d{,4}+$", "", route_or_segment)
     route_date = re.sub(r"^[^|]+\|", "", route)
 
     if format not in ["h264", "hevc"]: