mypy.

dmlc · Dec 18, 2024 · 8bc5b24 · 8bc5b24
1 parent 318b9b2
commit 8bc5b24
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/python/test_data_iterator.py b/tests/python/test_data_iterator.py
@@ -259,13 +259,13 @@ def test_cat_check() -> None:
     batches = []
 
     for i in range(n_batches):
-        X, y = tm.make_categorical(
+        X_df, y_arr = tm.make_categorical(
             n_samples=n_samples_per_batch,
             n_features=n_features,
             n_categories=3,
             onehot=False,
         )
-        batches.append((X, y))
+        batches.append((X_df, y_arr))
 
     X, y = list(zip(*batches))
     it = tm.IteratorForTest(X, y, None, cache=None, on_host=False)