Fix removed numpy NaN

GateNLP · Jul 2, 2024 · aabb618 · aabb618
1 parent 78204e5
commit aabb618
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/extractors/test_categories.py b/tests/extractors/test_categories.py
@@ -26,7 +26,7 @@ def test_equals_expected(datadir, categories_df):
 
 
 def test_parent_nan_replacement(categories_df):
-    assert categories_df["parent"].equals(ordered_col([np.NaN, np.NaN, 2]))
+    assert categories_df["parent"].equals(ordered_col([np.nan, np.nan, 2]))
 
 
 def test_link_locale(categories_df):