nvflare.lighter.poc_commands module

async_process(cmd_path, gpu_ids: List[int] | None = None)

clean_poc(poc_workspace: str)

client_gpu_assignments(clients: List[str], gpu_ids: List[int]) → Dict[str, List[int]]

def_poc_parser(sub_cmd)

get_cmd_path(poc_workspace, service_name, cmd)

get_gpu_ids(user_input_gpu_ids, host_gpu_ids) → List[int]

get_local_host_gpu_ids()

get_nvflare_home() → str

get_package_command(cmd_type: str, poc_workspace: str, package_dir) → str

get_stop_cmd(poc_workspace: str, service_dir_name: str)

get_upload_dir(poc_workspace: str) → str

handle_poc_cmd(cmd_args)

is_poc(cmd_args) → bool

is_poc_ready(poc_workspace: str)

prepare_env(gpu_ids: List[int] | None = None)

prepare_examples(poc_workspace: str)

prepare_poc(number_of_clients: int, poc_workspace: str)

sort_package_cmds(cmd_type, package_cmds: list) → list

start_poc(poc_workspace: str, gpu_ids: List[int], excluded=None, white_list=None)

stop_poc(poc_workspace: str, excluded=None, white_list=None)

sync_process(cmd_path)

validate_gpu_ids(gpu_ids: list, host_gpu_ids: list)

validate_poc_workspace(poc_workspace: str)