mirror of
https://github.com/IBM/ai-privacy-toolkit.git
synced 2026-05-07 11:02: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:
|
||||
self.features_names = dataset.features_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])]
|
||||
else:
|
||||
raise ValueError('No data provided')
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue