diff --git a/src/taggingbackends/data/dataset.py b/src/taggingbackends/data/dataset.py index 008a7af69cf265cbe33b17701304d34b10f455ae..595bd7974f9106be7d1595facc730105a84d513b 100644 --- a/src/taggingbackends/data/dataset.py +++ b/src/taggingbackends/data/dataset.py @@ -10,6 +10,7 @@ Torch-like dataset class for *larva_dataset hdf5* files. class LarvaDataset: def __init__(self, dataset=None, generator=None, subsets=(.8, .1, .1), balancing_strategy=None, class_weights=None): + self._path = None self.generator = generator self._full_set = dataset self.subset_shares = subsets @@ -26,7 +27,6 @@ class LarvaDataset: self.weight_classes = isinstance(balancing_strategy, str) and (balancing_strategy.lower() == 'auto') else: self.class_weights = class_weights - self._path = None """ *h5py.File*: *larva_dataset hdf5* file handler.