Skip to content
Snippets Groups Projects
Commit 3ba3e30f authored by Amir MOHAMMADI's avatar Amir MOHAMMADI
Browse files

Pass the same size as well to the lower function

parent 2de8c54a
Branches
Tags
1 merge request!103Propagated the option --allow-missing-files .....
Pipeline #
...@@ -305,8 +305,7 @@ def vstack_features(reader, paths, same_size=False, allow_missing_files=False): ...@@ -305,8 +305,7 @@ def vstack_features(reader, paths, same_size=False, allow_missing_files=False):
if same_size and allow_missing_files: if same_size and allow_missing_files:
raise ValueError("Both same_size and allow_missing_files cannot be True at" raise ValueError("Both same_size and allow_missing_files cannot be True at"
" the same time.") " the same time.")
iterable = _generate_features( iterable = _generate_features(reader, paths, same_size, allow_missing_files)
reader, paths, allow_missing_files=allow_missing_files)
dtype, shape = next(iterable) dtype, shape = next(iterable)
if same_size: if same_size:
total_size = int(len(paths) * numpy.prod(shape)) total_size = int(len(paths) * numpy.prod(shape))
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment