diff --git a/apt/minimization/minimizer.py b/apt/minimization/minimizer.py index 09b7099..2ab55bc 100644 --- a/apt/minimization/minimizer.py +++ b/apt/minimization/minimizer.py @@ -1107,4 +1107,4 @@ class GeneralizeToRepresentative(BaseEstimator, MetaEstimatorMixin, TransformerM to_remove.append(feature) for feature in to_remove: - del generalizations['categories'][feature] \ No newline at end of file + del generalizations['categories'][feature] diff --git a/tests/test_minimizer.py b/tests/test_minimizer.py index d022b47..be79d27 100644 --- a/tests/test_minimizer.py +++ b/tests/test_minimizer.py @@ -213,11 +213,11 @@ def test_minimizer_ncp(): gen2.transform(dataset=ad1) ncp6 = gen2.ncp - assert(ncp1 <= ncp3) - assert(ncp2 != ncp3) - assert(ncp3 != ncp4) - assert(ncp4 != ncp5) - assert(ncp6 == ncp4) + assert (ncp1 <= ncp3) + assert (ncp2 != ncp3) + assert (ncp3 != ncp4) + assert (ncp4 != ncp5) + assert (ncp6 == ncp4) def test_minimizer_ncp_categorical(): @@ -289,11 +289,11 @@ def test_minimizer_ncp_categorical(): gen2.transform(dataset=ad1) ncp6 = gen2.ncp - assert(ncp1 <= ncp3) - assert(ncp2 != ncp3) - assert(ncp3 != ncp4) - assert(ncp4 != ncp5) - assert(ncp6 == ncp4) + assert (ncp1 <= ncp3) + assert (ncp2 != ncp3) + assert (ncp3 != ncp4) + assert (ncp4 != ncp5) + assert (ncp6 == ncp4) def test_minimizer_fit_not_transform():