From 31e278fe60a32b8665648e33d0fea5357815558a Mon Sep 17 00:00:00 2001 From: olasaadi Date: Mon, 28 Mar 2022 00:07:54 +0300 Subject: [PATCH] fix bug --- apt/anonymization/anonymizer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/apt/anonymization/anonymizer.py b/apt/anonymization/anonymizer.py index db5a29b..e1b4a7e 100644 --- a/apt/anonymization/anonymizer.py +++ b/apt/anonymization/anonymizer.py @@ -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')