SYMBOL INDEX (564 symbols across 54 files)

FILE: xgboost_ray/callback.py
  class DistributedCallback (line 14) | class DistributedCallback(ABC):
    method on_init (line 29) | def on_init(self, actor: "RayXGBoostActor", *args, **kwargs):
    method before_data_loading (line 32) | def before_data_loading(
    method after_data_loading (line 37) | def after_data_loading(
    method before_train (line 42) | def before_train(self, actor: "RayXGBoostActor", *args, **kwargs):
    method after_train (line 45) | def after_train(self, actor: "RayXGBoostActor", result_dict: Dict, *ar...
    method before_predict (line 48) | def before_predict(self, actor: "RayXGBoostActor", *args, **kwargs):
    method after_predict (line 51) | def after_predict(
  class DistributedCallbackContainer (line 62) | class DistributedCallbackContainer:
    method __init__ (line 63) | def __init__(self, callbacks: Sequence[DistributedCallback]):
    method on_init (line 66) | def on_init(self, actor: "RayXGBoostActor", *args, **kwargs):
    method before_data_loading (line 70) | def before_data_loading(
    method after_data_loading (line 76) | def after_data_loading(
    method before_train (line 82) | def before_train(self, actor: "RayXGBoostActor", *args, **kwargs):
    method after_train (line 86) | def after_train(self, actor: "RayXGBoostActor", result_dict: Dict, *ar...
    method before_predict (line 90) | def before_predict(self, actor: "RayXGBoostActor", *args, **kwargs):
    method after_predict (line 94) | def after_predict(
  class EnvironmentCallback (line 105) | class EnvironmentCallback(DistributedCallback):
    method __init__ (line 106) | def __init__(self, env_dict: Dict[str, Any]):
    method on_init (line 109) | def on_init(self, actor, *args, **kwargs):

FILE: xgboost_ray/compat/__init__.py
  class TrainingCallback (line 12) | class TrainingCallback:
    method __init__ (line 13) | def __init__(self):
    method __call__ (line 22) | def __call__(self, callback_env: "xgb.core.CallbackEnv"):
    method before_training (line 37) | def before_training(self, model):
    method after_training (line 40) | def after_training(self, model):

FILE: xgboost_ray/compat/tracker.py
  class ExSocket (line 30) | class ExSocket(object):
    method __init__ (line 35) | def __init__(self, sock):
    method recvall (line 38) | def recvall(self, nbytes):
    method recvint (line 47) | def recvint(self):
    method sendint (line 50) | def sendint(self, n):
    method sendstr (line 53) | def sendstr(self, s):
    method recvstr (line 57) | def recvstr(self):
  function get_some_ip (line 66) | def get_some_ip(host):
  function get_host_ip (line 70) | def get_host_ip(hostIP=None):
  function get_family (line 94) | def get_family(addr):
  class SlaveEntry (line 98) | class SlaveEntry(object):
    method __init__ (line 99) | def __init__(self, sock, s_addr):
    method decide_rank (line 113) | def decide_rank(self, job_map):
    method assign_rank (line 120) | def assign_rank(self, rank, wait_conn, tree_map, parent_map, ring_map):
  class RabitTracker (line 178) | class RabitTracker(object):
    method __init__ (line 183) | def __init__(self, hostIP, nslave, port=9091, port_end=9999):
    method __del__ (line 203) | def __del__(self):
    method get_neighbor (line 207) | def get_neighbor(rank, nslave):
    method slave_envs (line 218) | def slave_envs(self):
    method get_tree (line 225) | def get_tree(self, nslave):
    method find_share_ring (line 233) | def find_share_ring(self, tree_map, parent_map, r):
    method get_ring (line 252) | def get_ring(self, tree_map, parent_map):
    method get_link_map (line 267) | def get_link_map(self, nslave):
    method accept_slaves (line 294) | def accept_slaves(self, nslave):
    method start (line 368) | def start(self, nslave):
    method join (line 375) | def join(self):
    method alive (line 379) | def alive(self):

FILE: xgboost_ray/data_sources/_distributed.py
  function get_actor_rank_ips (line 10) | def get_actor_rank_ips(actors: Sequence[ActorHandle]) -> Dict[int, str]:
  function assign_partitions_to_actors (line 24) | def assign_partitions_to_actors(

FILE: xgboost_ray/data_sources/csv.py
  class CSV (line 9) | class CSV(DataSource):
    method is_data_type (line 16) | def is_data_type(data: Any, filetype: Optional[RayFileType] = None) ->...
    method get_filetype (line 20) | def get_filetype(data: Any) -> Optional[RayFileType]:
    method load_data (line 26) | def load_data(
    method get_n (line 46) | def get_n(data: Any):

FILE: xgboost_ray/data_sources/dask.py
  function _assert_dask_installed (line 24) | def _assert_dask_installed():
  function ensure_ray_dask_initialized (line 37) | def ensure_ray_dask_initialized(
  class Dask (line 45) | class Dask(DataSource):
    method is_data_type (line 59) | def is_data_type(data: Any, filetype: Optional[RayFileType] = None) ->...
    method load_data (line 69) | def load_data(
    method convert_to_series (line 97) | def convert_to_series(data: Any) -> pd.Series:
    method get_actor_shards (line 114) | def get_actor_shards(
    method get_n (line 128) | def get_n(data: Any):
  function get_ip_to_parts (line 136) | def get_ip_to_parts(data: Any) -> Dict[int, Sequence[Any]]:

FILE: xgboost_ray/data_sources/data_source.py
  class RayFileType (line 13) | class RayFileType(Enum):
  class DataSource (line 22) | class DataSource:
    method is_data_type (line 40) | def is_data_type(data: Any, filetype: Optional[RayFileType] = None) ->...
    method get_filetype (line 57) | def get_filetype(data: Any) -> Optional[RayFileType]:
    method load_data (line 73) | def load_data(
    method update_feature_names (line 96) | def update_feature_names(matrix: "xgb.DMatrix", feature_names: Optiona...
    method convert_to_series (line 108) | def convert_to_series(data: Any) -> pd.Series:
    method get_column (line 119) | def get_column(
    method get_n (line 133) | def get_n(data: Any):
    method get_actor_shards (line 138) | def get_actor_shards(

FILE: xgboost_ray/data_sources/modin.py
  function _assert_modin_installed (line 33) | def _assert_modin_installed():
  class Modin (line 48) | class Modin(DataSource):
    method is_data_type (line 62) | def is_data_type(data: Any, filetype: Optional[RayFileType] = None) ->...
    method load_data (line 72) | def load_data(
    method convert_to_series (line 100) | def convert_to_series(data: Any) -> pd.Series:
    method get_actor_shards (line 114) | def get_actor_shards(
    method get_n (line 138) | def get_n(data: Any):

FILE: xgboost_ray/data_sources/numpy.py
  class Numpy (line 13) | class Numpy(DataSource):
    method is_data_type (line 17) | def is_data_type(data: Any, filetype: Optional[RayFileType] = None) ->...
    method update_feature_names (line 21) | def update_feature_names(matrix: "xgb.DMatrix", feature_names: Optiona...
    method load_data (line 26) | def load_data(

FILE: xgboost_ray/data_sources/object_store.py
  class ObjectStore (line 11) | class ObjectStore(DataSource):
    method is_data_type (line 15) | def is_data_type(data: Any, filetype: Optional[RayFileType] = None) ->...
    method load_data (line 21) | def load_data(
    method convert_to_series (line 35) | def convert_to_series(data: Any) -> pd.Series:

FILE: xgboost_ray/data_sources/pandas.py
  class Pandas (line 8) | class Pandas(DataSource):
    method is_data_type (line 12) | def is_data_type(data: Any, filetype: Optional[RayFileType] = None) ->...
    method load_data (line 16) | def load_data(

FILE: xgboost_ray/data_sources/parquet.py
  class Parquet (line 9) | class Parquet(DataSource):
    method is_data_type (line 16) | def is_data_type(data: Any, filetype: Optional[RayFileType] = None) ->...
    method get_filetype (line 20) | def get_filetype(data: Any) -> Optional[RayFileType]:
    method load_data (line 26) | def load_data(
    method get_n (line 47) | def get_n(data: Any):

FILE: xgboost_ray/data_sources/partitioned.py
  class Partitioned (line 18) | class Partitioned(DataSource):
    method is_data_type (line 33) | def is_data_type(data: Any, filetype: Optional[RayFileType] = None) ->...
    method load_data (line 37) | def load_data(
    method get_actor_shards (line 67) | def get_actor_shards(
    method get_n (line 97) | def get_n(data: Any):

FILE: xgboost_ray/data_sources/petastorm.py
  function _assert_petastorm_installed (line 15) | def _assert_petastorm_installed():
  class Petastorm (line 27) | class Petastorm(DataSource):
    method is_data_type (line 41) | def is_data_type(data: Any, filetype: Optional[RayFileType] = None) ->...
    method get_filetype (line 45) | def get_filetype(data: Any) -> Optional[RayFileType]:
    method load_data (line 66) | def load_data(
    method get_n (line 88) | def get_n(data: Any):

FILE: xgboost_ray/data_sources/ray_dataset.py
  function _assert_ray_data_available (line 20) | def _assert_ray_data_available():
  class RayDataset (line 32) | class RayDataset(DataSource):
    method is_data_type (line 40) | def is_data_type(data: Any, filetype: Optional[RayFileType] = None) ->...
    method load_data (line 47) | def load_data(
    method convert_to_series (line 73) | def convert_to_series(
    method get_actor_shards (line 87) | def get_actor_shards(
    method get_n (line 106) | def get_n(data: "ray.data.dataset.Dataset"):

FILE: xgboost_ray/elastic.py
  function _maybe_schedule_new_actors (line 19) | def _maybe_schedule_new_actors(
  function _update_scheduled_actor_states (line 98) | def _update_scheduled_actor_states(training_state: _TrainingState):
  function _get_actor_alive_status (line 145) | def _get_actor_alive_status(

FILE: xgboost_ray/examples/create_test_data.py
  function main (line 4) | def main():

FILE: xgboost_ray/examples/higgs.py
  function download_higgs (line 9) | def download_higgs(target_file):
  function main (line 29) | def main():

FILE: xgboost_ray/examples/higgs_parquet.py
  function csv_to_parquet (line 15) | def csv_to_parquet(in_file, out_file, chunksize=100_000, **csv_kwargs):
  function main (line 43) | def main():

FILE: xgboost_ray/examples/readme.py
  function readme_simple (line 4) | def readme_simple():
  function readme_predict (line 29) | def readme_predict():
  function readme_tune (line 45) | def readme_tune():

FILE: xgboost_ray/examples/readme_sklearn_api.py
  function readme_sklearn_api (line 1) | def readme_sklearn_api():

FILE: xgboost_ray/examples/simple.py
  function main (line 10) | def main(cpus_per_actor, num_actors):

FILE: xgboost_ray/examples/simple_dask.py
  function main (line 11) | def main(cpus_per_actor, num_actors):

FILE: xgboost_ray/examples/simple_modin.py
  function main (line 11) | def main(cpus_per_actor, num_actors):

FILE: xgboost_ray/examples/simple_objectstore.py
  function main (line 10) | def main(cpus_per_actor, num_actors):

FILE: xgboost_ray/examples/simple_partitioned.py
  class AnActor (line 14) | class AnActor:
    method genData (line 20) | def genData(self, rank, nranks, nrows):
  class Parted (line 36) | class Parted:
    method __init__ (line 39) | def __init__(self, parted):
  function main (line 43) | def main(cpus_per_actor, num_actors):

FILE: xgboost_ray/examples/simple_predict.py
  function main (line 10) | def main():

FILE: xgboost_ray/examples/simple_ray_dataset.py
  function main (line 11) | def main(cpus_per_actor, num_actors):

FILE: xgboost_ray/examples/simple_tune.py
  function train_breast_cancer (line 13) | def train_breast_cancer(config, ray_params):
  function main (line 39) | def main(cpus_per_actor, num_actors, num_samples):

FILE: xgboost_ray/examples/train_on_test_data.py
  function main (line 16) | def main(fname, num_actors=2):

FILE: xgboost_ray/examples/train_with_ml_dataset.py
  function main (line 18) | def main(fname, num_actors=2):

FILE: xgboost_ray/main.py
  class EarlyStopException (line 24) | class EarlyStopException(XGBoostError):
  function PublicAPI (line 69) | def PublicAPI(f):
  function _get_environ (line 110) | def _get_environ(item: str, old_val: Any):
  class _XGBoostEnv (line 129) | class _XGBoostEnv:
    method __getattribute__ (line 154) | def __getattribute__(self, item):
  class RayXGBoostTrainingError (line 178) | class RayXGBoostTrainingError(RuntimeError):
  class RayXGBoostTrainingStopped (line 185) | class RayXGBoostTrainingStopped(RuntimeError):
  class RayXGBoostActorAvailable (line 192) | class RayXGBoostActorAvailable(RuntimeError):
  function _assert_ray_support (line 199) | def _assert_ray_support():
  function _in_ray_tune_session (line 207) | def _in_ray_tune_session() -> bool:
  function _maybe_print_legacy_warning (line 213) | def _maybe_print_legacy_warning():
  function _is_client_connected (line 218) | def _is_client_connected() -> bool:
  class _RabitTrackerCompatMixin (line 225) | class _RabitTrackerCompatMixin:
    method accept_workers (line 228) | def accept_workers(self, n_workers: int):
    method worker_envs (line 231) | def worker_envs(self):
  class _RabitTracker (line 235) | class _RabitTracker(RabitTracker, _RabitTrackerCompatMixin):
    method start (line 242) | def start(self, nworker):
  function _start_rabit_tracker (line 256) | def _start_rabit_tracker(num_workers: int):
  function _stop_rabit_tracker (line 286) | def _stop_rabit_tracker(rabit_process: multiprocessing.Process):
  class _RabitContextBase (line 292) | class _RabitContextBase:
    method __init__ (line 302) | def __init__(self, actor_id: int, args: dict):
  class _RabitContext (line 310) | class _RabitContext(_RabitContextBase, CommunicatorContext):
    method __init__ (line 316) | def __init__(self, actor_id: int, args: dict):
    method __enter__ (line 320) | def __enter__(self):
    method __exit__ (line 323) | def __exit__(self, *args):
  class _RabitContext (line 315) | class _RabitContext(_RabitContextBase):
    method __init__ (line 316) | def __init__(self, actor_id: int, args: dict):
    method __enter__ (line 320) | def __enter__(self):
    method __exit__ (line 323) | def __exit__(self, *args):
  function _ray_get_actor_cpus (line 327) | def _ray_get_actor_cpus():
  function _ray_get_cluster_cpus (line 342) | def _ray_get_cluster_cpus():
  function _get_min_node_cpus (line 346) | def _get_min_node_cpus():
  function _set_omp_num_threads (line 355) | def _set_omp_num_threads():
  function _prepare_dmatrix_params (line 365) | def _prepare_dmatrix_params(param: Dict) -> Dict:
  function _get_dmatrix (line 379) | def _get_dmatrix(data: RayDMatrix, param: Dict) -> xgb.DMatrix:
  class RayParams (line 450) | class RayParams:
    method get_tune_resources (line 492) | def get_tune_resources(self):
  class _Checkpoint (line 508) | class _Checkpoint:
  function _validate_ray_params (line 513) | def _validate_ray_params(ray_params: Union[None, RayParams, dict]) -> Ra...
  class RayXGBoostActor (line 544) | class RayXGBoostActor:
    method __init__ (line 564) | def __init__(
    method set_queue (line 594) | def set_queue(self, queue: Queue):
    method set_stop_event (line 598) | def set_stop_event(self, stop_event: Event):
    method _get_stop_event (line 601) | def _get_stop_event(self):
    method pid (line 604) | def pid(self):
    method ip (line 608) | def ip(self):
    method _save_checkpoint_callback (line 612) | def _save_checkpoint_callback(self):
    method _stop_callback (line 628) | def _stop_callback(self):
    method load_data (line 654) | def load_data(self, data: RayDMatrix):
    method train (line 672) | def train(
    method predict (line 795) | def predict(self, model: xgb.Booster, data: RayDMatrix, **kwargs):
  class _RemoteRayXGBoostActor (line 814) | class _RemoteRayXGBoostActor(RayXGBoostActor):
  class _PrepareActorTask (line 818) | class _PrepareActorTask(MultiActorTask):
    method __init__ (line 819) | def __init__(
  function _autodetect_resources (line 835) | def _autodetect_resources(
  function _create_actor (line 862) | def _create_actor(
  function _trigger_data_load (line 895) | def _trigger_data_load(actor, dtrain, evals):
  function _handle_queue (line 902) | def _handle_queue(queue: Queue, checkpoint: _Checkpoint, callback_return...
  function _shutdown (line 925) | def _shutdown(
  function _create_placement_group (line 958) | def _create_placement_group(
  function _create_communication_processes (line 990) | def _create_communication_processes(added_tune_callback: bool = False):
  function _validate_kwargs_for_func (line 1022) | def _validate_kwargs_for_func(kwargs: Dict[str, Any], func: Callable, fu...
  class _TrainingState (line 1039) | class _TrainingState:
  function _train (line 1061) | def _train(
  function train (line 1341) | def train(
  function _predict (line 1750) | def _predict(model: xgb.Booster, data: RayDMatrix, ray_params: RayParams...
  function predict (line 1810) | def predict(

FILE: xgboost_ray/matrix.py
  class RayDataset (line 39) | class RayDataset:
  function concat_dataframes (line 65) | def concat_dataframes(dfs: List[Optional[pd.DataFrame]]):
  function ensure_sorted_by_qid (line 70) | def ensure_sorted_by_qid(
  class RayShardingMode (line 106) | class RayShardingMode(Enum):
  class RayDataIter (line 128) | class RayDataIter(DataIter):
    method __init__ (line 129) | def __init__(
    method __len__ (line 163) | def __len__(self):
    method reset (line 166) | def reset(self):
    method _prop (line 169) | def _prop(self, ref):
    method next (line 179) | def next(self, input_data: Callable):
  class _RayDMatrixLoader (line 199) | class _RayDMatrixLoader:
    method __init__ (line 200) | def __init__(
    method get_data_source (line 262) | def get_data_source(self) -> Type[DataSource]:
    method assert_enough_shards_for_actors (line 265) | def assert_enough_shards_for_actors(self, num_actors: int):
    method update_matrix_properties (line 270) | def update_matrix_properties(self, matrix: "xgb.DMatrix"):
    method assign_shards_to_actors (line 274) | def assign_shards_to_actors(self, actors: Sequence[ActorHandle]) -> bool:
    method _split_dataframe (line 283) | def _split_dataframe(
    method load_data (line 360) | def load_data(
  class _CentralRayDMatrixLoader (line 366) | class _CentralRayDMatrixLoader(_RayDMatrixLoader):
    method get_data_source (line 369) | def get_data_source(self) -> Type[DataSource]:
    method load_data (line 431) | def load_data(
  class _DistributedRayDMatrixLoader (line 490) | class _DistributedRayDMatrixLoader(_RayDMatrixLoader):
    method get_data_source (line 493) | def get_data_source(self) -> Type[DataSource]:
    method assert_enough_shards_for_actors (line 576) | def assert_enough_shards_for_actors(self, num_actors: int):
    method assign_shards_to_actors (line 595) | def assign_shards_to_actors(self, actors: Sequence[ActorHandle]) -> bool:
    method load_data (line 614) | def load_data(
  class RayDMatrix (line 697) | class RayDMatrix:
    method __init__ (line 787) | def __init__(
    method has_label (line 891) | def has_label(self):
    method assign_shards_to_actors (line 894) | def assign_shards_to_actors(self, actors: Sequence[ActorHandle]) -> bool:
    method assert_enough_shards_for_actors (line 900) | def assert_enough_shards_for_actors(self, num_actors: int):
    method load_data (line 903) | def load_data(self, num_actors: Optional[int] = None, rank: Optional[i...
    method get_data (line 936) | def get_data(
    method unload_data (line 954) | def unload_data(self):
    method update_matrix_properties (line 961) | def update_matrix_properties(self, matrix: "xgb.DMatrix"):
    method __hash__ (line 964) | def __hash__(self):
    method __eq__ (line 967) | def __eq__(self, other):
  class RayQuantileDMatrix (line 971) | class RayQuantileDMatrix(RayDMatrix):
  class RayDeviceQuantileDMatrix (line 977) | class RayDeviceQuantileDMatrix(RayDMatrix):
    method __init__ (line 980) | def __init__(
    method get_data (line 1024) | def get_data(
  function _can_load_distributed (line 1036) | def _can_load_distributed(source: Data) -> bool:
  function _detect_distributed (line 1063) | def _detect_distributed(source: Data) -> bool:
  function _get_sharding_indices (line 1088) | def _get_sharding_indices(
  function combine_data (line 1114) | def combine_data(sharding: RayShardingMode, data: Iterable) -> np.ndarray:

FILE: xgboost_ray/session.py
  class RayXGBoostSession (line 8) | class RayXGBoostSession:
    method __init__ (line 9) | def __init__(self, rank: int, queue: Optional[Queue]):
    method get_actor_rank (line 13) | def get_actor_rank(self):
    method set_queue (line 16) | def set_queue(self, queue):
    method put_queue (line 19) | def put_queue(self, item):
  function init_session (line 33) | def init_session(*args, **kwargs):
  function get_session (line 44) | def get_session() -> RayXGBoostSession:
  function set_session_queue (line 56) | def set_session_queue(queue: Queue):
  function get_actor_rank (line 62) | def get_actor_rank() -> int:
  function get_rabit_rank (line 68) | def get_rabit_rank() -> int:
  function put_queue (line 79) | def put_queue(*args, **kwargs):

FILE: xgboost_ray/sklearn.py
  function _deprecate_positional_args (line 56) | def _deprecate_positional_args(f):
  function _wrap_evaluation_matrices (line 72) | def _wrap_evaluation_matrices(
  function _cls_predict_proba (line 176) | def _cls_predict_proba(n_classes: int, prediction, vstack: Callable):
  function _get_doc (line 223) | def _get_doc(object: Any) -> Optional[str]:
  function _treat_estimator_doc (line 239) | def _treat_estimator_doc(doc: Optional[str]) -> Optional[str]:
  function _treat_X_doc (line 253) | def _treat_X_doc(doc: Optional[str]) -> Optional[str]:
  function _xgboost_version_warn (line 268) | def _xgboost_version_warn(f):
  function _check_if_params_are_ray_dmatrix (line 280) | def _check_if_params_are_ray_dmatrix(
  class RayXGBMixin (line 338) | class RayXGBMixin:
    method _ray_set_ray_params_n_jobs (line 341) | def _ray_set_ray_params_n_jobs(
    method _ray_predict (line 357) | def _ray_predict(
    method _ray_get_wrap_evaluation_matrices_compat_kwargs (line 392) | def _ray_get_wrap_evaluation_matrices_compat_kwargs(
    method _configure_fit (line 418) | def _configure_fit(
    method _set_evaluation_result (line 442) | def _set_evaluation_result(self, evals_result) -> None:
  class RayXGBRegressor (line 451) | class RayXGBRegressor(XGBRegressor, RayXGBMixin):
    method fit (line 455) | def fit(
    method _can_use_inplace_predict (line 564) | def _can_use_inplace_predict(self) -> bool:
    method predict (line 567) | def predict(
    method load_model (line 593) | def load_model(self, fname):
  class RayXGBRFRegressor (line 602) | class RayXGBRFRegressor(RayXGBRegressor):
    method __init__ (line 607) | def __init__(self, *args, **kwargs):
    method __init__ (line 614) | def __init__(
    method get_xgb_params (line 631) | def get_xgb_params(self):
    method get_num_boosting_rounds (line 636) | def get_num_boosting_rounds(self):
  class RayXGBClassifier (line 644) | class RayXGBClassifier(XGBClassifier, RayXGBMixin):
    method fit (line 648) | def fit(
    method _can_use_inplace_predict (line 795) | def _can_use_inplace_predict(self) -> bool:
    method predict (line 798) | def predict(
    method predict_proba (line 839) | def predict_proba(
    method load_model (line 867) | def load_model(self, fname):
  class RayXGBRFClassifier (line 880) | class RayXGBRFClassifier(RayXGBClassifier):
    method __init__ (line 884) | def __init__(self, *args, **kwargs):
    method __init__ (line 891) | def __init__(
    method get_xgb_params (line 908) | def get_xgb_params(self):
    method get_num_boosting_rounds (line 913) | def get_num_boosting_rounds(self):
  class RayXGBRanker (line 921) | class RayXGBRanker(XGBRanker, RayXGBMixin):
    method fit (line 925) | def fit(
    method _can_use_inplace_predict (line 1048) | def _can_use_inplace_predict(self) -> bool:
    method predict (line 1051) | def predict(
    method load_model (line 1077) | def load_model(self, fname):

FILE: xgboost_ray/tests/conftest.py
  function get_default_fixure_system_config (line 14) | def get_default_fixure_system_config():
  function get_default_fixture_ray_kwargs (line 24) | def get_default_fixture_ray_kwargs():
  function _ray_start_cluster (line 37) | def _ray_start_cluster(**kwargs):
  function ray_start_cluster (line 69) | def ray_start_cluster(request):

FILE: xgboost_ray/tests/fault_tolerance.py
  class FaultToleranceManager (line 15) | class FaultToleranceManager:
    method __init__ (line 16) | def __init__(self, start_boost_round: int = 0):
    method schedule_kill (line 29) | def schedule_kill(self, rank: int, boost_round: int):
    method delay_return (line 33) | def delay_return(self, rank: int, start_boost_round: int, end_boost_ro...
    method inc_boost_round (line 37) | def inc_boost_round(self, rank: int):
    method log_iteration (line 42) | def log_iteration(self, rank: int, boost_round: int):
    method should_die (line 46) | def should_die(self, rank: int):
    method should_sleep (line 56) | def should_sleep(self, rank: int):
    method get_logs (line 64) | def get_logs(self):
  class DelayedLoadingCallback (line 68) | class DelayedLoadingCallback(DistributedCallback):
    method __init__ (line 71) | def __init__(self, ft_manager: ActorHandle, reload_data=True, sleep_ti...
    method after_data_loading (line 76) | def after_data_loading(self, actor, data, *args, **kwargs):
  class DieCallback (line 83) | class DieCallback(TrainingCallback):
    method __init__ (line 89) | def __init__(self, ft_manager: ActorHandle, training_delay: float = 0):
    method before_iteration (line 94) | def before_iteration(self, model, epoch, evals_log):
    method after_iteration (line 103) | def after_iteration(self, model, epoch, evals_log):

FILE: xgboost_ray/tests/release/benchmark_cpu_gpu.py
  function train_ray (line 22) | def train_ray(

FILE: xgboost_ray/tests/release/benchmark_ft.py
  function train_ray (line 32) | def train_ray(
  function ft_setup (line 160) | def ft_setup(
  function run_experiments (line 191) | def run_experiments(config, files, aws):

FILE: xgboost_ray/tests/release/custom_objective_metric.py
  class XGBoostDistributedAPITest (line 6) | class XGBoostDistributedAPITest(XGBoostAPITest):
    method _init_ray (line 7) | def _init_ray(self):

FILE: xgboost_ray/tests/release/tune_placement.py
  class PlacementCallback (line 47) | class PlacementCallback(TrainingCallback):
    method before_training (line 50) | def before_training(self, model):
    method after_iteration (line 55) | def after_iteration(self, model, epoch, evals_log):
  function tune_test (line 62) | def tune_test(

FILE: xgboost_ray/tests/test_client.py
  function start_client_server_4_cpus (line 11) | def start_client_server_4_cpus():
  function start_client_server_5_cpus (line 18) | def start_client_server_5_cpus():
  function start_client_server_5_cpus_modin (line 25) | def start_client_server_5_cpus_modin(monkeypatch):
  function test_simple_train (line 32) | def test_simple_train(start_client_server_4_cpus):
  function test_simple_tune (line 40) | def test_simple_tune(start_client_server_4_cpus):
  function test_simple_dask (line 47) | def test_simple_dask(start_client_server_5_cpus):
  function test_simple_modin (line 54) | def test_simple_modin(start_client_server_5_cpus_modin):
  function test_client_actor_cpus (line 61) | def test_client_actor_cpus(start_client_server_5_cpus):
  function test_simple_ray_dataset (line 88) | def test_simple_ray_dataset(start_client_server_5_cpus):

FILE: xgboost_ray/tests/test_colocation.py
  class _MockQueueActor (line 17) | class _MockQueueActor(_QueueActor):
    method get_node_id (line 18) | def get_node_id(self):
  class _MockEventActor (line 22) | class _MockEventActor(_EventActor):
    method get_node_id (line 23) | def get_node_id(self):
  class TestColocation (line 28) | class TestColocation(unittest.TestCase):
    method setUp (line 29) | def setUp(self) -> None:
    method tearDown (line 59) | def tearDown(self) -> None:
    method test_communication_colocation (line 66) | def test_communication_colocation(self):
    method test_no_tune_spread (line 104) | def test_no_tune_spread(self):
    method test_tune_pack (line 139) | def test_tune_pack(self):
    method test_timeout (line 191) | def test_timeout(self):

FILE: xgboost_ray/tests/test_data_source.py
  class _DistributedDataSourceTest (line 16) | class _DistributedDataSourceTest:
    method setUp (line 17) | def setUp(self):
    method tearDown (line 24) | def tearDown(self) -> None:
    method _init_ray (line 28) | def _init_ray(self):
    method _testAssignPartitions (line 32) | def _testAssignPartitions(self, part_nodes, actor_nodes, expected_acto...
    method _testDataSourceAssignment (line 35) | def _testDataSourceAssignment(self, part_nodes, actor_nodes, expected_...
    method testAssignEvenTrivial (line 38) | def testAssignEvenTrivial(self):
    method testAssignEvenRedistributeOne (line 56) | def testAssignEvenRedistributeOne(self):
    method testAssignEvenRedistributeMost (line 75) | def testAssignEvenRedistributeMost(self):
    method testAssignUnevenTrivial (line 96) | def testAssignUnevenTrivial(self):
    method testAssignUnevenRedistribute (line 112) | def testAssignUnevenRedistribute(self):
    method testAssignUnevenRedistributeColocated (line 129) | def testAssignUnevenRedistributeColocated(self):
    method testAssignUnevenRedistributeAll (line 146) | def testAssignUnevenRedistributeAll(self):
  class ModinDataSourceTest (line 167) | class ModinDataSourceTest(_DistributedDataSourceTest, unittest.TestCase):
    method _testAssignPartitions (line 170) | def _testAssignPartitions(self, part_nodes, actor_nodes, expected_acto...
    method _getActorToParts (line 190) | def _getActorToParts(self, actors_to_node, node_to_part):
    method _testDataSourceAssignment (line 209) | def _testDataSourceAssignment(self, part_nodes, actor_nodes, expected_...
  class DaskDataSourceTest (line 295) | class DaskDataSourceTest(_DistributedDataSourceTest, unittest.TestCase):
    method _testAssignPartitions (line 298) | def _testAssignPartitions(self, part_nodes, actor_nodes, expected_acto...
    method _getActorToParts (line 320) | def _getActorToParts(self, actors_to_node, node_to_part):
    method _testDataSourceAssignment (line 344) | def _testDataSourceAssignment(self, part_nodes, actor_nodes, expected_...
  class PartitionedSourceTest (line 439) | class PartitionedSourceTest(_DistributedDataSourceTest, unittest.TestCase):
    method _testAssignPartitions (line 440) | def _testAssignPartitions(self, part_nodes, actor_nodes, expected_acto...
    method _mk_partitioned (line 463) | def _mk_partitioned(self, part_to_node, nr, nc, shapes):
    method _getActorToParts (line 490) | def _getActorToParts(self, actors_to_node, partitions, part_to_node, p...
    method _testDataSourceAssignment (line 508) | def _testDataSourceAssignment(self, part_nodes, actor_nodes, expected_...

FILE: xgboost_ray/tests/test_end_to_end.py
  function _make_callback (line 19) | def _make_callback(tmpdir: str) -> DistributedCallback:
  class XGBoostRayEndToEndTest (line 56) | class XGBoostRayEndToEndTest(unittest.TestCase):
    method setUp (line 71) | def setUp(self):
    method tearDown (line 92) | def tearDown(self):
    method testSingleTraining (line 96) | def testSingleTraining(self):
    method testHalfTraining (line 105) | def testHalfTraining(self):
    method test_client_actor_cpus (line 141) | def test_client_actor_cpus(self):
    method _testJointTraining (line 162) | def _testJointTraining(self, sharding=RayShardingMode.INTERLEAVED, sof...
    method testJointTrainingInterleaved (line 203) | def testJointTrainingInterleaved(self):
    method testJointTrainingBatch (line 208) | def testJointTrainingBatch(self):
    method testTrainPredict (line 213) | def testTrainPredict(
    method testTrainPredictSoftprob (line 256) | def testTrainPredictSoftprob(self):
    method testTrainPredictRemote (line 262) | def testTrainPredictRemote(self):
    method testTrainPredictClient (line 266) | def testTrainPredictClient(self):
    method testDistributedCallbacksTrainPredict (line 279) | def testDistributedCallbacksTrainPredict(self, init=True, remote=False):
    method testDistributedCallbacksTrainPredictClient (line 307) | def testDistributedCallbacksTrainPredictClient(self):
    method testFailPrintErrors (line 321) | def testFailPrintErrors(self):
    method testKwargsValidation (line 355) | def testKwargsValidation(self):
    method testRanking (line 374) | def testRanking(self):
    method testFeatureWeightsParam (line 429) | def testFeatureWeightsParam(self):

FILE: xgboost_ray/tests/test_fault_tolerance.py
  class _FakeTask (line 30) | class _FakeTask(MagicMock):
    method is_ready (line 33) | def is_ready(self):
  class XGBoostRayFaultToleranceTest (line 37) | class XGBoostRayFaultToleranceTest(unittest.TestCase):
    method setUp (line 43) | def setUp(self):
    method tearDown (line 79) | def tearDown(self) -> None:
    method testTrainingContinuationKilled (line 90) | def testTrainingContinuationKilled(self):
    method testTrainingContinuationElasticKilled (line 125) | def testTrainingContinuationElasticKilled(self):
    method testTrainingContinuationElasticKilledRestarted (line 169) | def testTrainingContinuationElasticKilledRestarted(self):
    method testTrainingContinuationElasticMultiKilled (line 223) | def testTrainingContinuationElasticMultiKilled(self):
    method testTrainingContinuationElasticFailed (line 255) | def testTrainingContinuationElasticFailed(self):
    method testTrainingStop (line 297) | def testTrainingStop(self):
    method testTrainingStopElastic (line 309) | def testTrainingStopElastic(self):
    method testCheckpointContinuationValidity (line 340) | def testCheckpointContinuationValidity(self):
    method testSameResultWithAndWithoutError (line 401) | def testSameResultWithAndWithoutError(self):
    method testMaybeScheduleNewActors (line 454) | def testMaybeScheduleNewActors(self):
    method testFaultToleranceManager (line 587) | def testFaultToleranceManager(self):

FILE: xgboost_ray/tests/test_matrix.py
  class XGBoostRayDMatrixTest (line 21) | class XGBoostRayDMatrixTest(unittest.TestCase):
    method setUp (line 24) | def setUp(self):
    method setUpClass (line 47) | def setUpClass(cls):
    method tearDownClass (line 51) | def tearDownClass(cls):
    method testSameObject (line 54) | def testSameObject(self):
    method testColumnOrdering (line 64) | def testColumnOrdering(self):
    method _testMatrixCreation (line 74) | def _testMatrixCreation(self, in_x, in_y, multi_label=False, **kwargs):
    method testFromNumpy (line 115) | def testFromNumpy(self):
    method testFromPandasDfDf (line 120) | def testFromPandasDfDf(self):
    method testFromPandasDfSeries (line 125) | def testFromPandasDfSeries(self):
    method testFromPandasDfString (line 130) | def testFromPandasDfString(self):
    method testFromModinDfDf (line 135) | def testFromModinDfDf(self):
    method testFromModinDfSeries (line 148) | def testFromModinDfSeries(self):
    method testFromModinDfString (line 161) | def testFromModinDfString(self):
    method testFromDaskDfSeries (line 175) | def testFromDaskDfSeries(self):
    method testFromDaskDfArray (line 189) | def testFromDaskDfArray(self):
    method testFromDaskDfString (line 204) | def testFromDaskDfString(self):
    method testFromPetastormParquetString (line 219) | def testFromPetastormParquetString(self):
    method testFromPetastormMultiParquetString (line 236) | def testFromPetastormMultiParquetString(self):
    method testFromCSVString (line 261) | def testFromCSVString(self):
    method testFromMultiCSVString (line 273) | def testFromMultiCSVString(self):
    method testFromParquetStringMultiLabel (line 294) | def testFromParquetStringMultiLabel(self):
    method testFromParquetString (line 310) | def testFromParquetString(self):
    method testFromMultiParquetStringMultiLabel (line 321) | def testFromMultiParquetStringMultiLabel(self):
    method testFromMultiParquetString (line 343) | def testFromMultiParquetString(self):
    method testDetectDistributed (line 364) | def testDetectDistributed(self):
    method testTooManyActorsDistributed (line 393) | def testTooManyActorsDistributed(self):
    method testTooManyActorsCentral (line 399) | def testTooManyActorsCentral(self):
    method testBatchShardingAllActorsGetIndices (line 406) | def testBatchShardingAllActorsGetIndices(self):
    method testLegacyParams (line 411) | def testLegacyParams(self):
    method testFeatureWeightsParam (line 440) | def testFeatureWeightsParam(self):
    method testQidSortedBehaviorXGBoost (line 451) | def testQidSortedBehaviorXGBoost(self):
    method testQidSortedParquet (line 473) | def testQidSortedParquet(self):

FILE: xgboost_ray/tests/test_sklearn.py
  function softmax (line 53) | def softmax(x):
  function softprob_obj (line 58) | def softprob_obj(classes):
  function get_basescore (line 81) | def get_basescore(model: xgb.XGBModel) -> float:
  class TemporaryDirectory (line 91) | class TemporaryDirectory(object):
    method __enter__ (line 94) | def __enter__(self):
    method __exit__ (line 98) | def __exit__(self, exc_type, exc_value, traceback):
  class XGBoostRaySklearnTest (line 102) | class XGBoostRaySklearnTest(unittest.TestCase):
    method setUp (line 103) | def setUp(self):
    method tearDown (line 107) | def tearDown(self) -> None:
    method _init_ray (line 111) | def _init_ray(self):
    method run_binary_classification (line 115) | def run_binary_classification(self, cls, ray_dmatrix_params=None):
    method test_binary_classification (line 143) | def test_binary_classification(self):
    method test_binary_classification_dmatrix_params (line 146) | def test_binary_classification_dmatrix_params(self):
    method test_binary_rf_classification (line 156) | def test_binary_rf_classification(self):
    method test_multiclass_classification (line 159) | def test_multiclass_classification(self):
    method test_stacking_regression (line 210) | def test_stacking_regression(self):
    method test_stacking_classification (line 231) | def test_stacking_classification(self):
    method test_select_feature (line 262) | def test_select_feature(self):
    method test_num_parallel_tree (line 277) | def test_num_parallel_tree(self):
    method test_california_housing_regression (line 315) | def test_california_housing_regression(self):
    method run_california_housing_rf_regression (line 343) | def run_california_housing_rf_regression(self, tree_method):
    method test_california_housing_rf_regression (line 358) | def test_california_housing_rf_regression(self):
    method test_parameter_tuning (line 363) | def test_parameter_tuning(self):
    method test_regression_with_custom_objective (line 383) | def test_regression_with_custom_objective(self):
    method test_classification_with_custom_objective (line 419) | def test_classification_with_custom_objective(self):
    method test_sklearn_api (line 472) | def test_sklearn_api(self):
    method test_sklearn_api_gblinear (line 493) | def test_sklearn_api_gblinear(self):
    method test_sklearn_random_state (line 518) | def test_sklearn_random_state(self):
    method test_sklearn_n_jobs (line 535) | def test_sklearn_n_jobs(self):
    method test_parameters_access (line 548) | def test_parameters_access(self):
    method test_kwargs_error (line 574) | def test_kwargs_error(self):
    method test_kwargs_grid_search (line 582) | def test_kwargs_grid_search(self):
    method test_sklearn_clone (line 603) | def test_sklearn_clone(self):
    method test_sklearn_get_default_params (line 616) | def test_sklearn_get_default_params(self):
    method test_validation_weights_xgbmodel (line 634) | def test_validation_weights_xgbmodel(self):
    method test_validation_weights_xgbclassifier (line 711) | def test_validation_weights_xgbclassifier(self):
    method save_load_model (line 768) | def save_load_model(self, model_path):
    method test_save_load_model (line 808) | def test_save_load_model(self):
    method test_XGBClassifier_resume (line 913) | def test_XGBClassifier_resume(self):
    method test_constraint_parameters (line 957) | def test_constraint_parameters(self):
    method test_pandas_input (line 1057) | def test_pandas_input(self):
    method run_boost_from_prediction (line 1155) | def run_boost_from_prediction(self, tree_method):
    method boost_from_prediction (line 1187) | def boost_from_prediction(self, tree_method):
    method test_boost_from_prediction_hist (line 1196) | def test_boost_from_prediction_hist(self):
    method test_boost_from_prediction_approx (line 1203) | def test_boost_from_prediction_approx(self):
    method test_boost_from_prediction_exact (line 1208) | def test_boost_from_prediction_exact(self):
    method test_estimator_type (line 1216) | def test_estimator_type(self):
    method test_ranking (line 1240) | def test_ranking(self):

FILE: xgboost_ray/tests/test_sklearn_matrix.py
  class XGBoostRaySklearnMatrixTest (line 17) | class XGBoostRaySklearnMatrixTest(unittest.TestCase):
    method setUp (line 18) | def setUp(self):
    method tearDown (line 23) | def tearDown(self) -> None:
    method _init_ray (line 27) | def _init_ray(self):
    method testClassifierNoLabelEncoder (line 34) | def testClassifierNoLabelEncoder(self, n_class=2):
    method testClassifierMulticlassNoLabelEncoder (line 73) | def testClassifierMulticlassNoLabelEncoder(self):
    method testRegressor (line 76) | def testRegressor(self):

FILE: xgboost_ray/tests/test_tune.py
  class XGBoostRayTuneTest (line 19) | class XGBoostRayTuneTest(unittest.TestCase):
    method setUp (line 20) | def setUp(self):
    method tearDown (line 71) | def tearDown(self):
    method testNumIters (line 77) | def testNumIters(self):
    method testNumItersClient (line 104) | def testNumItersClient(self):
    method testPlacementOptions (line 116) | def testPlacementOptions(self):
    method testElasticFails (line 127) | def testElasticFails(self):
    method testReplaceTuneCheckpoints (line 138) | def testReplaceTuneCheckpoints(self):
    method testEndToEndCheckpointing (line 153) | def testEndToEndCheckpointing(self):
    method testEndToEndCheckpointingOrigTune (line 170) | def testEndToEndCheckpointingOrigTune(self):

FILE: xgboost_ray/tests/test_xgboost_api.py
  function gradient (line 16) | def gradient(predt: np.ndarray, dtrain: xgb.DMatrix) -> np.ndarray:
  function hessian (line 21) | def hessian(predt: np.ndarray, dtrain: xgb.DMatrix) -> np.ndarray:
  function squared_log (line 26) | def squared_log(
  function rmsle (line 35) | def rmsle(predt: np.ndarray, dtrain: xgb.DMatrix) -> Tuple[str, float]:
  class XGBoostAPITest (line 42) | class XGBoostAPITest(unittest.TestCase):
    method setUp (line 45) | def setUp(self):
    method tearDown (line 69) | def tearDown(self) -> None:
    method _init_ray (line 73) | def _init_ray(self):
    method testCustomObjectiveFunction (line 77) | def testCustomObjectiveFunction(self):
    method testCustomMetricFunction (line 104) | def testCustomMetricFunction(self):
    method testCallbacks (line 154) | def testCallbacks(self):

FILE: xgboost_ray/tests/utils.py
  function get_num_trees (line 15) | def get_num_trees(bst: xgb.Booster):
  function create_data (line 22) | def create_data(num_rows: int, num_cols: int, dtype: np.dtype = np.float...
  function create_labels (line 31) | def create_labels(
  function create_parquet (line 47) | def create_parquet(
  function create_parquet_in_tempdir (line 74) | def create_parquet_in_tempdir(
  function flatten_obj (line 93) | def flatten_obj(obj: Union[List, Dict], keys=None, base=None):
  function tree_obj (line 107) | def tree_obj(bst: xgb.Booster):
  function _kill_callback (line 111) | def _kill_callback(die_lock_file: str, actor_rank: int = 0, fail_iterati...
  function _fail_callback (line 145) | def _fail_callback(die_lock_file: str, actor_rank: int = 0, fail_iterati...
  function _checkpoint_callback (line 179) | def _checkpoint_callback(frequency: int = 1, before_iteration_=False):
  function _sleep_callback (line 205) | def _sleep_callback(sleep_iteration: int = 6, sleep_seconds: int = 5):

FILE: xgboost_ray/tune.py
  class TuneReportCheckpointCallback (line 26) | class TuneReportCheckpointCallback(OrigTuneReportCheckpointCallback):
    method after_iteration (line 27) | def after_iteration(self, model, epoch: int, evals_log: Dict):
    method after_training (line 41) | def after_training(self, model):
  class TuneReportCallback (line 51) | class TuneReportCallback(OrigTuneReportCallback):
    method __new__ (line 52) | def __new__(cls: type, *args, **kwargs):
  function _try_add_tune_callback (line 60) | def _try_add_tune_callback(kwargs: Dict):
  function _get_tune_resources (line 107) | def _get_tune_resources(
  function load_model (line 130) | def load_model(model_path):

FILE: xgboost_ray/util.py
  class Unavailable (line 9) | class Unavailable:
    method __init__ (line 12) | def __init__(self):
  class _EventActor (line 16) | class _EventActor:
    method __init__ (line 17) | def __init__(self):
    method set (line 20) | def set(self):
    method clear (line 23) | def clear(self):
    method is_set (line 26) | def is_set(self):
  class Event (line 31) | class Event:
    method __init__ (line 32) | def __init__(self, actor_options: Optional[Dict] = None):
    method set (line 36) | def set(self):
    method clear (line 39) | def clear(self):
    method is_set (line 42) | def is_set(self):
    method shutdown (line 45) | def shutdown(self):
  class MultiActorTask (line 52) | class MultiActorTask:
    method __init__ (line 62) | def __init__(self, pending_futures: Optional[List[ray.ObjectRef]] = No...
    method is_ready (line 66) | def is_ready(self):
  function get_current_node_resource_key (line 82) | def get_current_node_resource_key() -> str:
  function force_on_current_node (line 100) | def force_on_current_node(task_or_actor):