mirror of
https://github.com/vale981/ray
synced 2025-03-09 12:56:46 -04:00
8804758409
* Add XGBoost release tests * Add more xgboost release tests * Use failure state manager * Add release test documentation * Fix wording * Automate fault tolerance tests
26 lines
642 B
Python
26 lines
642 B
Python
"""Distributed XGBoost API test
|
|
|
|
This test runs unit tests on a distributed cluster. This will confirm that
|
|
XGBoost API features like custom metrics/objectives work with remote
|
|
trainables.
|
|
|
|
Test owner: krfricke
|
|
|
|
Acceptance criteria: Unit tests should pass (requires pytest).
|
|
"""
|
|
|
|
import ray
|
|
|
|
from xgboost_ray.tests.test_xgboost_api import XGBoostAPITest
|
|
|
|
|
|
class XGBoostDistributedAPITest(XGBoostAPITest):
|
|
def _init_ray(self):
|
|
if not ray.is_initialized():
|
|
ray.init(address="auto")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
import pytest
|
|
import sys
|
|
sys.exit(pytest.main(["-v", f"{__file__}::XGBoostDistributedAPITest"]))
|