FIX typo in variables sent to function

rapidsai · Oct 3, 2024 · 39bc5f9 · 39bc5f9
1 parent 8eaef40
commit 39bc5f9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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: