nvflare.fuel.utils.function_utils module

find_task_fn(task_fn_path) Callable[source]

Return function given a function path.

Parameters:

task_fn_path (str) – function path

Returns:

function

ex: train.main -> main

custom/train.main -> main custom.train.main -> main

require_arguments(func)[source]

Inspect function to get required arguments.

Parameters:

func – function

Returns:

require_args (bool), args_size (int), args_default_size (int)