From 4e68c447e65aae7038ecbf0adc8acb1b41533a8a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Pawe=C5=82=20Czy=C5=BC?= <pczyz@protonmail.com>
Date: Mon, 27 May 2024 10:54:25 +0200
Subject: [PATCH] Fix type error

---
 labelshift/datasets/split.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/labelshift/datasets/split.py b/labelshift/datasets/split.py
index 3bc33d9..f2f64cc 100644
--- 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 {