Handle exception when importing RetroGPTChunkDatasets (NVIDIA#11415)

* Handle exception when importing RetroGPTChunkDatasets Signed-off-by: Guyue Huang <[email protected]> * Update baseline Signed-off-by: guyueh1 <[email protected]> * Revert "Update baseline" This reverts commit 21a4734. * Update baseline Signed-off-by: guyueh1 <[email protected]> * Revert "Update baseline" This reverts commit 8d35fcc. --------- Signed-off-by: Guyue Huang <[email protected]> Signed-off-by: guyueh1 <[email protected]> Co-authored-by: guyueh1 <[email protected]>
paarthneekhara · Jan 18, 2025 · f22663d · f22663d
1 parent 9db1ee7
commit f22663d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nemo/collections/nlp/data/language_modeling/megatron/retro_dataset.py b/nemo/collections/nlp/data/language_modeling/megatron/retro_dataset.py
@@ -47,7 +47,7 @@
 
     HAVE_TE_AND_MEGATRON_CORE = True
 
-except (ImportError, ModuleNotFoundError):
+except (ImportError, ModuleNotFoundError, Exception):
 
     HAVE_TE_AND_MEGATRON_CORE = False
     from typing import Any