[xgb] re-enable xgboost_ray tests (#13416)

* re-enable

* fix

* update xgb_ray version

.travis.yml

@@ -389,7 +389,6 @@ matrix:
         # - ./ci/keep_alive bazel test --config=ci $(./scripts/bazel_export_options) --build_tests_only --test_tag_filters=-tf,-pytorch,-py37 python/ray/util/sgd/...
         - ./ci/keep_alive bazel test --config=ci $(./scripts/bazel_export_options) --build_tests_only --test_tag_filters=tf,-pytorch,-py37 python/ray/util/sgd/...
         - ./ci/keep_alive bazel test --config=ci $(./scripts/bazel_export_options) --build_tests_only --test_tag_filters=-tf,pytorch,-py37 python/ray/util/sgd/...
-        # - ./ci/keep_alive bazel test --config=ci $(./scripts/bazel_export_options) --build_tests_only python/ray/util/xgboost/...
 
     # Docs: Tests and examples.
     - os: linux
@@ -418,6 +417,7 @@ matrix:
         - . ./ci/travis/ci.sh build
       script:
         - ./ci/keep_alive bazel test --config=ci $(./scripts/bazel_export_options) --build_tests_only --test_tag_filters=py37 python/ray/tune/...
+        - ./ci/keep_alive bazel test --config=ci $(./scripts/bazel_export_options) --build_tests_only python/ray/util/xgboost/...
         # There are no python 3.7 tests for RaySGD at the moment
         # - ./ci/keep_alive bazel test --config=ci --build_tests_only --test_tag_filters=py37 python/ray/util/sgd/...
         # - ./ci/keep_alive bazel test --config=ci --build_tests_only --test_tag_filters=py37 doc/...

python/ray/util/xgboost/BUILD

@@ -1,26 +1,28 @@
 # --------------------------------------------------------------------
-# Tests from the python/ray/util/sgd/tests directory.
+# Tests from the python/ray/util/xgboost directory.
 # Please keep these sorted alphabetically.
 # --------------------------------------------------------------------
-# py_test(
-#     name = "simple_example",
-#     size = "small",
-#     srcs = ["simple_example.py"],
-#     deps = [":xgb_lib"],
-#     tags = ["exclusive"],
-# )
+py_test(
+ name = "simple_example",
+ size = "small",
+ srcs = ["simple_example.py"],
+ deps = [":xgb_lib"],
+ tags = ["exclusive"],
+)
+
+py_test(
+ name = "simple_tune",
+ size="small",
+ srcs = ["simple_tune.py"],
+ deps = [":xgb_lib"],
+ tags = ["exclusive"]
+)
+
+# This is a dummy test dependency that causes the above tests to be
+# re-run if any of these files changes.
+py_library(
+ name = "xgb_lib",
+ srcs = glob(["**/*.py"]),
+)
 
-# py_test(
-#     name = "simple_tune",
-#     size="small",
-#     srcs = ["simple_tune.py"],
-#     deps = [":xgb_lib"],
-#     tags = ["exlcusive"]
-# )
 
-# # This is a dummy test dependency that causes the above tests to be
-# # re-run if any of these files changes.
-# py_library(
-#     name = "xgb_lib",
-#     srcs = glob(["**/*.py"], exclude=["tests/*.py"]),
-# )

python/ray/util/xgboost/simple_example.py

@@ -31,7 +31,7 @@ def main():
         train_set,
         evals=[(test_set, "eval")],
         evals_result=evals_result,
-        ray_params=RayParams(max_actor_restarts=1),
+        ray_params=RayParams(max_actor_restarts=1, num_actors=1),
         verbose_eval=False)
 
     bst.save_model("simple.xgb")

python/requirements/requirements_upstream.txt

@@ -3,4 +3,4 @@
 # So we separate its own requirements file.
 
 tune-sklearn==0.2.1
-xgboost_ray==0.0.1
+xgboost_ray==0.0.2