From 805c6f182a8cab6a82b1ba8d200c99553371694f Mon Sep 17 00:00:00 2001
From: Shriya Palsamudram <spalsamudram@nvidia.com>
Date: Wed, 23 Oct 2024 12:57:55 -0700
Subject: [PATCH] Fix failing unit tests

Signed-off-by: Shriya Palsamudram <spalsamudram@nvidia.com>
---
 tests/lightning/test_dist_ckpt.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/lightning/test_dist_ckpt.py b/tests/lightning/test_dist_ckpt.py
index d5037f0aa5731..886b1085ed55e 100644
--- a/tests/lightning/test_dist_ckpt.py
+++ b/tests/lightning/test_dist_ckpt.py
@@ -34,6 +34,7 @@ def set_env():
 def _get_strategy():
     strategy = nl.MegatronStrategy(
         enable_nemo_ckpt_io=False,
+        ckpt_async_save=False,
     )
     return strategy