Update python/cudf/cudf_pandas_tests/third_party_integration_tests/te…

…sts/test_xgboost.py
rapidsai · Dec 18, 2024 · a368262 · a368262
1 parent 11d37f7
commit a368262
Showing 1 changed file with 0 additions and 2 deletions.
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
@@ -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)