From cd7e2138b7919e0796db7e7902bf49b20065f4f8 Mon Sep 17 00:00:00 2001
From: Samuel Hoffman <hoffman.sc@gmail.com>
Date: Mon, 9 Dec 2024 22:08:33 -0500
Subject: [PATCH] Replace `np.NaN`s (#544)

use `np.nan` instead
---
 tests/sklearn/test_datasets.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/sklearn/test_datasets.py b/tests/sklearn/test_datasets.py
index 3ef86172..a092530e 100644
--- 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)