Fix deprecations with pandas 2.2

sebp · Feb 1, 2024 · 0c1e1e0 · 0c1e1e0
1 parent 93a1e2a
commit 0c1e1e0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/sksurv/column.py b/sksurv/column.py
@@ -199,7 +199,7 @@ def transform(column):
             except ValueError:
                 classes = column.dropna().unique()
                 classes.sort(kind="mergesort")
-                nc = column.replace(classes, np.arange(classes.shape[0], dtype=np.int64))
+                nc = column.map(dict(zip(classes, range(classes.shape[0]))))
             return nc
         if column.dtype == bool:
             return column.astype(np.int64)

diff --git a/tests/test_column.py b/tests/test_column.py
@@ -42,7 +42,7 @@ def non_numeric_data(self):
             }
         )
 
-        data.loc[:, "q3"] = data.loc[:, "q3"].astype("category")
+        data["q3"] = data.loc[:, "q3"].astype("category")
         return data
 
     def data_numeric(self):