Fix PDF reading (#2141)

Co-authored-by: Nikita Manovich <nikita.manovich@intel.com>
cvat-ai · Sep 9, 2020 · 8d3c95a · 8d3c95a
1 parent 81edf0b
commit 8d3c95a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -27,6 +27,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>)
 - Fixed CVAT format import for frame stepped tasks (<https://github.com/openvinotoolkit/cvat/pull/2151>)
 
 ### 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())