fix

Signed-off-by: Hemil Desai <hemild@nvidia.com>

Author: hemildesai

tests/test_suites/llm/grpo-moonlight-16b-automodel-1n8g-ep8.sh

@@ -39,4 +39,7 @@ if [[ $(jq 'to_entries | .[] | select(.key == "train/loss") | .value | keys | ma
         'data["train/reward"]["30"] > 0.4' \
         'data["train/grad_norm"]["30"] < 0.2' \
         'data["train/grad_norm"]["30"] > 0.1'
+
+    # Clean up checkpoint directory after successful run to save space.
+    rm -rf "$CKPT_DIR"
 fi