apunkt/ai-privacy-toolkit
1
0
Fork 0
mirror of https://github.com/IBM/ai-privacy-toolkit.git synced 2026-06-26 15:49:37 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

formatting

Browse source 
Signed-off-by: abigailt <abigailt@il.ibm.com>
This commit is contained in:
abigailt 2023-08-10 13:20:28 +03:00
parent f85fc87bdd
commit 69e45d99e5
4 changed files with 17 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

3
tests/test_minimizer.py
View file

@ -246,6 +246,7 @@ def create_encoder(numeric_features, categorical_features, x):
return preprocessor, encoded
def test_minimizer_params_not_transform(get_cells):
# Assume two features, age and height, and boolean label
cells, features, x, y = get_cells
@ -385,7 +386,7 @@ def test_minimizer_fit_not_transform(get_data_two_features):
gen.fit(dataset=train_dataset, generalize_using_transform=False)
gener = gen.generalizations
expected_generalizations = {'ranges': {'age': [], 'height':[157.0]}, 'categories': {}, 'untouched': []}
expected_generalizations = {'ranges': {'age': [], 'height': [157.0]}, 'categories': {}, 'untouched': []}
compare_generalizations(gener, expected_generalizations)