diff options
Diffstat (limited to 'data/hdf5.py')
| -rw-r--r-- | data/hdf5.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/data/hdf5.py b/data/hdf5.py index 6e2f9a4..f3d6da2 100644 --- a/data/hdf5.py +++ b/data/hdf5.py @@ -13,7 +13,7 @@ class TaxiDataset(H5PYDataset): def __init__(self, which_set, filename='data.hdf5', **kwargs): self.filename = filename kwargs.setdefault('load_in_memory', True) - super(TaxiDataset, self).__init__(self.data_path, which_set, **kwargs) + super(TaxiDataset, self).__init__(self.data_path, (which_set,), **kwargs) @property def data_path(self): |