Skip to content

Commit 1fe62a3

Browse files
authored
Merge pull request #38 from KumarLabJax/allow-mp4
Allow mp4 videos
2 parents 202cc44 + 45fa83f commit 1fe62a3

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)