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
No related branches found
No related tags found
1 merge request!103Propagated the option --allow-missing-files .....
Pipeline #
Hide whitespace changes
Inline Side-by-side
bob/bio/base/utils/io.py
+ 1
− 2
View file @ 3ba3e30f
...@@ -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% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment