Fix PDF reading

cvat-ai · Sep 7, 2020 · 1beea09 · 1beea09
1 parent f43863a
commit 1beea09
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cvat/apps/engine/task.py b/cvat/apps/engine/task.py
@@ -311,7 +311,7 @@ def update_progress(progress):
                     counter_ = itertools.count()
                     if isinstance(extractor, MEDIA_TYPES['archive']['extractor']):
                         media_files = [os.path.relpath(path, upload_dir) for path in extractor._source_path]
-                    elif isinstance(extractor, MEDIA_TYPES['zip']['extractor']):
+                elif isinstance(extractor, (MEDIA_TYPES['zip']['extractor'], MEDIA_TYPES['pdf']['extractor'])):
                         media_files = extractor._source_path
 
                     numbers_sequence = range(db_data.start_frame, min(data['stop_frame'] if data['stop_frame'] else len(media_files), len(media_files)), db_data.get_frame_step())