-
Notifications
You must be signed in to change notification settings - Fork 80
Description
Bug description
In connectomics/data/dataset/build.py
, line 351
the input loading works the following way:
volume, label, valid_mask = _get_input(cfg, mode, rank, dir_name_init, img_name_init, min_size=sample_volume_size)
However, if we look at the function _get_input
definition:
def _get_input(cfg, mode='train', rank=None, preload_data=None, dir_name_init: Optional[list] = None, img_name_init: Optional[list] = None, min_size: Optional[tuple] = None, image_only_test: bool = True):
Which means that in line 351
, dir_name_init
will be used as preload_data
in the _get_input
function
Bug fix proposition
Changing line 351
by adding a keyword for dir_name_init
:
volume, label, valid_mask = _get_input(cfg, mode, rank, dir_name_init=dir_name_init, img_name_init=img_name_init, min_size=sample_volume_size)