2
2
#
3
3
# SPDX-License-Identifier: MIT
4
4
5
- from unittest import mock
6
5
from typing import Optional , Callable
6
+ from unittest import mock
7
7
8
8
from datumaro .components .errors import DatasetNotFoundError
9
9
@@ -20,8 +20,7 @@ def updated_find_sources_recursive(
20
20
max_depth : int = 3 ,
21
21
):
22
22
sources = super (importer , cls )._find_sources_recursive (
23
- path , ext , extractor_name ,
24
- filename , dirname , file_filter , max_depth
23
+ path , ext , extractor_name , filename , dirname , file_filter , max_depth
25
24
)
26
25
27
26
if not sources :
@@ -43,6 +42,12 @@ def updated_generate_not_found_error(cls, path):
43
42
return updated_generate_not_found_error
44
43
45
44
def wrap_importer (importer ):
46
- mock .patch .object (importer , '_find_sources_recursive' , new = wrap_find_sources_recursive (importer )).start ()
47
- mock .patch .object (importer , '_generate_not_found_error' , new = wrap_generate_not_found_error ()).start ()
48
- mock .patch .object (importer , '_not_found_error_data' , new = {"ext" : "" , "filename" : "" }, create = True ).start ()
45
+ mock .patch .object (
46
+ importer , '_find_sources_recursive' , new = wrap_find_sources_recursive (importer )
47
+ ).start ()
48
+ mock .patch .object (
49
+ importer , '_generate_not_found_error' , new = wrap_generate_not_found_error ()
50
+ ).start ()
51
+ mock .patch .object (
52
+ importer , '_not_found_error_data' , new = {"ext" : "" , "filename" : "" }, create = True
53
+ ).start ()
0 commit comments