From c0393bb90f2f873c7df463d5d0d921c763557b6f Mon Sep 17 00:00:00 2001 From: Anton Alexander Date: Thu, 19 Feb 2026 15:19:36 +0000 Subject: [PATCH] Add is_async to CheckpointingConfig TypedDict The is_async parameter is already consumed by AutomodelCheckpointManager (automodel_checkpoint.py:118) and passed through in save_checkpoint (automodel_checkpoint.py:297), but was missing from the CheckpointingConfig TypedDict definition. This causes Hydra to reject `+checkpointing.is_async=true` as an unknown key. Signed-off-by: dmvevents --- nemo_rl/utils/checkpoint.py | 1 + 1 file changed, 1 insertion(+) 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: