Skip to content

Commit 8f09431

Browse files
committed
fix: Check file extensions if we are processing multiple file types
1 parent 85a2544 commit 8f09431

File tree

3 files changed

+5
-1
lines changed

3 files changed

+5
-1
lines changed

mapillary_tools/process_geotag_properties.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -289,6 +289,7 @@ def _process_videos_beta(vars_args: T.Dict):
289289
"num_processes": vars_args["num_processes"],
290290
"device_make": vars_args["device_make"],
291291
"device_model": vars_args["device_model"],
292+
"check_file_suffix": len(vars_args["filetypes"]) > 1,
292293
}
293294
extractor = VideoDataExtractor(options)
294295
return extractor.process()

mapillary_tools/video_data_extraction/cli_options.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,3 +20,4 @@ class CliOptions(T.TypedDict, total=False):
2020
num_processes: int
2121
device_make: T.Optional[str]
2222
device_model: T.Optional[str]
23+
check_file_suffix: bool

mapillary_tools/video_data_extraction/extract_video_data.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,9 @@ def __init__(self, options: CliOptions) -> None:
3333
def process(self) -> T.List[MetadataOrError]:
3434
paths = self.options["paths"]
3535
self._check_paths(paths)
36-
video_files = utils.find_videos(paths)
36+
video_files = utils.find_videos(
37+
paths, check_file_suffix=self.options["check_file_suffix"]
38+
)
3739
self._check_sources_cardinality(video_files)
3840

3941
num_processes = self.options["num_processes"] or None

0 commit comments

Comments
 (0)