Skip to content

Commit

Permalink
Merge pull request #38 from KumarLabJax/allow-mp4
Browse files Browse the repository at this point in the history
Allow mp4 videos
  • Loading branch information
SkepticRaven authored May 6, 2024
2 parents 202cc44 + 45fa83f commit 1fe62a3
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/project/project.py
Original file line number Diff line number Diff line change
Expand Up @@ -641,7 +641,7 @@ def counts(self, behavior):
@staticmethod
def get_videos(dir_path: Path):
""" Get list of video filenames (without path) in a directory """
return [f.name for f in dir_path.glob("*.avi")]
return [f.name for f in dir_path.glob("*") if f.suffix in ['.avi', '.mp4']]

def get_labeled_features(self, behavior=None, progress_callable=None):
"""
Expand Down

0 comments on commit 1fe62a3

Please sign in to comment.