nvflare.edge.simulation.devices.num module

class NumDevice(device_id, min_train_time=1.0, max_train_time=5.0)[source]

Bases: SimulatedDevice

do_task(task: TaskResponse) dict[source]
class NumDeviceFactory(min_train_time=1.0, max_train_time=5.0)[source]

Bases: DeviceFactory

make_device(device_id: str) SimulatedDevice[source]
class NumProcessor(min_train_time=1.0, max_train_time=5.0)[source]

Bases: DeviceTaskProcessor

process_task(task: TaskResponse) dict[source]

Process a task and return the result. This method is repeated for each task until all tasks are done

Parameters:

task – The task information from server

Returns:

The result as a dict

setup(job: JobResponse) None[source]

Setup for a new job

Args

device: the SimulatedDevice object job: Job information returned by server

shutdown() None[source]

Clean-up the resources allocated for the job and get ready for next