Handle exception when importing RetroGPTChunkDatasets (#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]>
NVIDIA · Dec 5, 2024 · 63dd8da · 63dd8da
1 parent 453a515
commit 63dd8da
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