pass a tuple

rapidsai · Nov 8, 2024 · ab03a4d · ab03a4d
1 parent 7ee0165
commit ab03a4d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/cudf/cudf_pandas_tests/third_party_integration_tests/tests/test_catboost.py b/python/cudf/cudf_pandas_tests/third_party_integration_tests/tests/test_catboost.py
@@ -113,7 +113,7 @@ def test_catboost_with_categorical_features():
     "X, y",
     [
         (
-            pd.DataFrame(rng.standard_normal(100, 5)),
+            pd.DataFrame(rng.standard_normal((100, 5))),
             pd.Series(rng.standard_normal(100)),
         ),
         (rng.standard_normal(100, 5), rng.standard_normal(100)),