Catching InputDataError (#6)

DiogenesAnalytics · Feb 6, 2024 · 84b9308 · 84b9308
1 parent 13696ed
commit 84b9308
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/frame_sampling/strategy.py b/src/frame_sampling/strategy.py
@@ -152,9 +152,14 @@ def sample(
                 # create dir
                 sample_subdir.mkdir(parents=True, exist_ok=exist_ok)
 
+            try:
                 # get frame samples from single video
                 self._sample_single_video(video_path, sample_subdir)
 
+            except av.InvalidDataError as error:
+                # decide how to handle error
+                self._handle_exceptions(error, video_path)
+
 
 class MinimalSampler(BaseSampler):
     """Simplest frame sampling strategy."""