Replace np.NaNs

use `np.nan` instead Signed-off-by: Samuel Hoffman <[email protected]>
Trusted-AI · Dec 9, 2024 · d48cdb6 · d48cdb6
1 parent e011686
commit d48cdb6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/sklearn/test_datasets.py b/tests/sklearn/test_datasets.py
@@ -17,7 +17,7 @@
     fetch_german, fetch_compas, fetch_lawschool_gpa, fetch_meps)
 
 
-df = pd.DataFrame([[1, 2, 3, 'a'], [5, 6, 7, 'b'], [np.NaN, 10, 11, 'c']],
+df = pd.DataFrame([[1, 2, 3, 'a'], [5, 6, 7, 'b'], [np.nan, 10, 11, 'c']],
                   columns=['X1', 'X2', 'y', 'Z'])
 basic = partial(standardize_dataset, df=df, prot_attr='Z', target='y',
                 dropna=False)