diff --git a/nemo_rl/utils/checkpoint.py b/nemo_rl/utils/checkpoint.py index 740e4cfc2d..49d2d3e2d7 100644 --- a/nemo_rl/utils/checkpoint.py +++ b/nemo_rl/utils/checkpoint.py @@ -65,6 +65,7 @@ class CheckpointingConfig(TypedDict): model_repo_id: NotRequired[str] # Default: "" is_peft: NotRequired[bool] # Default: False peft_config: NotRequired[Any] # Default: None + is_async: NotRequired[bool] # Default: False class CheckpointManager: