nvflare.apis.persistable module

class StatePersistable[source]

Bases: object

get_persist_state(fl_ctx: FLContext) dict[source]

Generate data from state to be persisted.

Parameters:

fl_ctx – FLContext

Returns:

A dict serializable persist data

restore(state_data: dict, fl_ctx: FLContext)[source]

Restore the state from persisted data.

Parameters:
  • state_data – serialized persist data

  • fl_ctx – FLContext