linter4

aimclub · Jan 9, 2024 · c25e578 · c25e578
1 parent 20ca87d
commit c25e578
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/tests/test_general/test_fairness.py b/tests/test_general/test_fairness.py
@@ -1,6 +1,5 @@
 import numpy as np
 import pandas as pd
-from sklearn.model_selection import train_test_split
 from lightgbm import LGBMClassifier
 
 from stable_gnn.fairness import Fair
@@ -63,13 +62,12 @@ def simple_splitter(arr):
 
 
 def test_fairness():
-
     dataset = prepare_data()
     cl = LGBMClassifier(verbose=-1)
     fairness = Fair(dataset, estimator=cl)
-    res = fairness.run(number_iterations=20, 
-                      interior_classifier='knn',
-                      multiplier=50)
+    res = fairness.run(number_iterations=20,
+                       interior_classifier='knn',
+                       multiplier=50)
 
     assert res["accuracy_of_initial_classifier"] - res["accuracy_of_fair_classifier"] <= 0.5
     assert res["fairness_of_fair_classifier_diff"] <= res["fairness_of_initial_classifier_diff"]