nvflare.app_opt.xgboost.histogram_based_v2.defs module

class Constant[source]

Bases: object

CONFIG_TASK_NAME = 'config'
CONFIG_TASK_TIMEOUT = 10
CONF_KEY_CLIENT_RANKS = 'client_ranks'
CONF_KEY_NUM_ROUNDS = 'num_rounds'
CONF_KEY_RANK = 'rank'
CONF_KEY_WORLD_SIZE = 'world_size'
DUMMY_BUFFER_SIZE = 4
ERR_INVALID_RANK = -2
ERR_NO_CLIENT_FOR_RANK = -3
ERR_OP_MISMATCH = -1
ERR_TARGET_ERROR = -4
EVENT_AFTER_ALL_GATHER_V = 'xgb.after_all_gather_v'
EVENT_AFTER_BROADCAST = 'xgb.after_broadcast'
EVENT_BEFORE_ALL_GATHER_V = 'xgb.before_all_gather_v'
EVENT_BEFORE_BROADCAST = 'xgb.before_broadcast'
EXIT_CODE_CANT_START = 101
HEADER_KEY_ENCRYPTED_DATA = 'xgb.encrypted_data'
HEADER_KEY_HORIZONTAL = 'xgb.horizontal'
HEADER_KEY_IN_AGGR = 'xgb.in_aggr'
HEADER_KEY_ORIGINAL_BUF_SIZE = 'xgb.original_buf_size'
JOB_STATUS_CHECK_INTERVAL = 2.0
MAX_CLIENT_OP_INTERVAL = 600.0
MSG_KEY_EXIT_CODE = 'exit_code'
MSG_KEY_XGB_OP = 'xgb.op'
MSG_KEY_XGB_REQ_ID = 'xgb.req_id'
MSG_KEY_XGB_REQ_RECEIVED = 'xgb.req_received'
MSG_KEY_XGB_REQ_TRY_NUM = 'xgb.req_try_num'
OPCODE_ALL_GATHER = 1
OPCODE_ALL_GATHER_V = 2
OPCODE_ALL_REDUCE = 3
OPCODE_BROADCAST = 4
OPCODE_DONE = 99
OPCODE_NONE = 0
OP_ALL_GATHER = 'all_gather'
OP_ALL_GATHER_V = 'all_gather_v'
OP_ALL_REDUCE = 'all_reduce'
OP_BROADCAST = 'broadcast'
PARAM_KEY_DATA_TYPE = 'xgb.data_type'
PARAM_KEY_EVENT = 'xgb.event'
PARAM_KEY_HEADERS = 'xgb.headers'
PARAM_KEY_RANK = 'xgb.rank'
PARAM_KEY_RCV_BUF = 'xgb.rcv_buf'
PARAM_KEY_REDUCE_OP = 'xgb.reduce_op'
PARAM_KEY_REPLY = 'xgb.reply'
PARAM_KEY_REQUEST = 'xgb.request'
PARAM_KEY_ROOT = 'xgb.root'
PARAM_KEY_SEND_BUF = 'xgb.send_buf'
PARAM_KEY_SEQ = 'xgb.seq'
RUNNER_CTX_CLIENT_NAME = 'client_name'
RUNNER_CTX_DATA_LOADER = 'data_loader'
RUNNER_CTX_MODEL_DIR = 'model_dir'
RUNNER_CTX_NUM_ROUNDS = 'num_rounds'
RUNNER_CTX_PORT = 'port'
RUNNER_CTX_RANK = 'rank'
RUNNER_CTX_SERVER_ADDR = 'server_addr'
RUNNER_CTX_TB_DIR = 'tb_dir'
RUNNER_CTX_WORLD_SIZE = 'world_size'
START_TASK_NAME = 'start'
START_TASK_TIMEOUT = 10
TASK_CHECK_INTERVAL = 0.5
TOPIC_CLIENT_DONE = 'xgb.client_done'
TOPIC_XGB_REQUEST = 'xgb.request'
TOPIC_XGB_REQUEST_CHECK = 'xgb.req_check'
WORKFLOW_PROGRESS_TIMEOUT = 3600.0
XGB_SERVER_READY_TIMEOUT = 10.0