mirror of
https://github.com/IBM/ai-privacy-toolkit.git
synced 2026-05-08 11:32:37 +02:00
fix bug
This commit is contained in:
parent
6172385d91
commit
31e278fe60
1 changed files with 1 additions and 1 deletions
|
|
@ -49,7 +49,7 @@ class Anonymize:
|
||||||
if dataset.features_names is not None:
|
if dataset.features_names is not None:
|
||||||
self.features_names = dataset.features_names
|
self.features_names = dataset.features_names
|
||||||
# if features is None, use numbers instead of names
|
# if features is None, use numbers instead of names
|
||||||
elif dataset.get_samples().shape[0] != 0:
|
elif dataset.get_samples().shape[1] != 0:
|
||||||
self.features_names = [i for i in range(dataset.get_samples().shape[1])]
|
self.features_names = [i for i in range(dataset.get_samples().shape[1])]
|
||||||
else:
|
else:
|
||||||
raise ValueError('No data provided')
|
raise ValueError('No data provided')
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue