Fix PDF reading

cvat-ai · Sep 8, 2020 · 9d1b2f2 · 9d1b2f2
1 parent f43863a
commit 9d1b2f2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -25,6 +25,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ### Fixed
 - Fixed multiple errors which arises when polygon is of length 5 or less (<https://github.com/opencv/cvat/pull/2100>)
+- Fixed task creation from PDF (<https://github.com/opencv/cvat/pull/2141>)
 
 ### Security
 -

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())