fix: new pandas behavior

Author: jlwalke2

tests/conftest.py

@@ -494,8 +494,8 @@ def cancer_dataset():
     raw = datasets.load_breast_cancer()
     df = pd.DataFrame(raw.data, columns=raw.feature_names)
     df["Type"] = raw.target
-    df.Type = df.Type.astype("category")
-    df.Type.cat.categories = raw.target_names
+    df.Type = df.Type.astype("category").cat.rename_categories(raw.target_names)
+
     return df
 
 

tests/unit/test_model_parameters.py

@@ -35,8 +35,7 @@ def train_data():
     iris = pd.DataFrame(raw.data, columns=raw.feature_names)
     iris = iris.join(pd.DataFrame(raw.target))
     iris.columns = ["SepalLength", "SepalWidth", "PetalLength", "PetalWidth", "Species"]
-    iris["Species"] = iris["Species"].astype("category")
-    iris.Species.cat.categories = raw.target_names
+    iris["Species"] = iris["Species"].astype("category").cat.rename_categories(raw.target_names)
     return iris.iloc[:, 0:4], iris["Species"]