diff --git a/fluents/dataset.py b/fluents/dataset.py index 4446dc8..488f97f 100644 --- a/fluents/dataset.py +++ b/fluents/dataset.py @@ -473,11 +473,11 @@ def read_ftsv(fd): # Create dataset of specified type if type == 'category': - ds = CategoryDataset(matrix, dims) + ds = CategoryDataset(matrix, dims, name) elif type == 'network': - ds = GraphDataset(matrix, dims) + ds = GraphDataset(matrix, dims, name) else: - ds = Dataset(matrix, dims) + ds = Dataset(matrix, dims, name) return ds