Fix failing unit tests

Signed-off-by: Shriya Palsamudram <[email protected]>

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