flip.nvflare.components.persist_and_cleanup

Classes

PersistToS3AndCleanup

Module Contents

class flip.nvflare.components.persist_and_cleanup.PersistToS3AndCleanup(model_id: str = '', persistor_id: str = AppConstants.DEFAULT_PERSISTOR_ID, flip: flip.FLIP = FLIP())

Bases: nvflare.apis.fl_component.FLComponent

_model_id_fallback = ''

_model_id: str | None = None

persistor_id: str

model_persistor: nvflare.app_opt.pt.file_model_persistor.PTFileModelPersistor | None = None

model_inventory: dict

model_dir: str = ''

bucket_name: str = ''

flip

_resolve_model_id(fl_ctx: nvflare.apis.fl_context.FLContext) → str

Resolve model ID lazily from job metadata, falling back to the constructor arg.

Parameters:: fl_ctx (FLContext) – The FL context for the current job.
Returns:: The resolved model ID.
Return type:: str

execute(fl_ctx: nvflare.apis.fl_context.FLContext) → None

upload_results_to_s3_bucket(fl_ctx: nvflare.apis.fl_context.FLContext) → None: Uploads the final aggregated model and reports to an S3 bucket as a zip file.

cleanup(fl_ctx: nvflare.apis.fl_context.FLContext) → None: Cleans up the workspace by deleting the transfer and save directories for the model ID.