Fix type error

pawel-czyz · May 27, 2024 · 4e68c44 · 4e68c44
1 parent e5df39b
commit 4e68c44
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/labelshift/datasets/split.py b/labelshift/datasets/split.py
@@ -73,7 +73,7 @@ def split_dataset(
 
     if set(np.unique(dataset.target)) != set(range(n_labels)):
         raise ValueError(
-            f"Labels must be 0-indexed integers: {dataset.target_names} != "
+            f"Labels must be 0-indexed integers: {dataset.target} != "
             f"{set(range(n_labels))}."
         )
     if {