From 39bc5f9707c8c35be240953fa7629d822f253d5a Mon Sep 17 00:00:00 2001 From: Dante Gama Dessavre Date: Wed, 2 Oct 2024 19:08:25 -0500 Subject: [PATCH] FIX typo in variables sent to function --- python/cuml/cuml/model_selection/_split.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/cuml/cuml/model_selection/_split.py b/python/cuml/cuml/model_selection/_split.py index 6c4a406d97..44e857ef02 100644 --- a/python/cuml/cuml/model_selection/_split.py +++ b/python/cuml/cuml/model_selection/_split.py @@ -406,10 +406,10 @@ def _process_df_objs( return df_train, df_test X_train, X_test = _process_df_objs( - X, x_type, X_train, X_test, train_indices + X, x_type, X_train, X_test, train_indices, test_indices ) y_train, y_test = _process_df_objs( - y, y_type, y_train, y_test, test_indices + y, y_type, y_train, y_test, train_indices, test_indices ) else: