diff --git a/python/cudf/cudf_pandas_tests/third_party_integration_tests/tests/test_xgboost.py b/python/cudf/cudf_pandas_tests/third_party_integration_tests/tests/test_xgboost.py index 9dfc09da863..ba98273404d 100644 --- a/python/cudf/cudf_pandas_tests/third_party_integration_tests/tests/test_xgboost.py +++ b/python/cudf/cudf_pandas_tests/third_party_integration_tests/tests/test_xgboost.py @@ -127,8 +127,6 @@ def test_predict(device: str) -> np.ndarray: # After https://github.com/dmlc/xgboost/pull/11014, .inplace_predict() # returns a real cupy array when called on a cudf.pandas proxy dataframe. # So we need to ensure we have a valid numpy array. - # TODO: We should remove the call to .get() when .inplace_predict() - # returns a cudf.pandas proxy numpy array if not isinstance(predt1, np.ndarray): predt1 = predt1.get() np.testing.assert_allclose(predt0, predt1)