Skip to content

Commit 8f7fda9

Browse files
committed
Allowing mp4 videos alongside avi
1 parent 70b6e55 commit 8f7fda9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/project/project.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -641,7 +641,7 @@ def counts(self, behavior):
641641
@staticmethod
642642
def get_videos(dir_path: Path):
643643
""" Get list of video filenames (without path) in a directory """
644-
return [f.name for f in dir_path.glob("*.avi")]
644+
return [f.name for f in dir_path.glob("*") if f.suffix in ['.avi', '.mp4']]
645645

646646
def get_labeled_features(self, behavior=None, progress_callable=None):
647647
"""

0 commit comments

Comments
 (0)