Numpy 2.0 removes np.NaN. (#1376)

IBM · Jul 25, 2024 · 7fd8cc7 · 7fd8cc7
1 parent 165a4c0
commit 7fd8cc7
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/lale/datasets/uci/uci_datasets.py b/lale/datasets/uci/uci_datasets.py
@@ -107,7 +107,7 @@ def fetch_drugscom():
                 {"description": "drugName", "type": "string"},
                 {
                     "description": "condition",
-                    "anyOf": [{"type": "string"}, {"enum": [np.NaN]}],
+                    "anyOf": [{"type": "string"}, {"enum": [np.nan]}],
                 },
                 {"description": "review", "type": "string"},
                 {

diff --git a/lale/lib/xgboost/xgb_classifier.py b/lale/lib/xgboost/xgb_classifier.py
@@ -789,10 +789,10 @@ def score(self, X, y):
                     "type": "number",
                 },
                 {
-                    "enum": [None, np.NaN],
+                    "enum": [None, np.nan],
                 },
             ],
-            "default": np.NaN,
+            "default": np.nan,
             "description": "Value in the data which needs to be present as a missing value. If"
             " If None, defaults to np.nan.",
         },

diff --git a/lale/lib/xgboost/xgb_regressor.py b/lale/lib/xgboost/xgb_regressor.py
@@ -714,10 +714,10 @@ def score(self, X, y):
                     "type": "number",
                 },
                 {
-                    "enum": [None, np.NaN],
+                    "enum": [None, np.nan],
                 },
             ],
-            "default": np.NaN,
+            "default": np.nan,
             "description": "Value in the data which needs to be present as a missing value. If"
             " If None, defaults to np.nan.",
         },