Commit ec7f1e17 authored by Amir MOHAMMADI's avatar Amir MOHAMMADI
Browse files

make sure the biofiles are kept

parent f83a7087
......@@ -57,7 +57,7 @@ class BioGenerator(object):
return -1
self.database = database
self.biofiles = biofiles
self.biofiles = list(biofiles)
self.load_data = load_data
self.biofile_to_label = biofile_to_label
self.multiple_samples = multiple_samples
......@@ -77,7 +77,7 @@ class BioGenerator(object):
self._output_shapes = (
data.shape, tf.TensorShape([]), tf.TensorShape([]))
logger.debug("Initializing a dataset with %d files and %s types ""Initializing a dataset with %d files and %s types "
"and %s shapes", len(self.biofiles), self.output_types,
Markdown is supported
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