relax exception type

aws · Aug 10, 2022 · d9c4492 · d9c4492
1 parent 122e429
commit d9c4492
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/sagemaker_training/smdataparallel.py b/src/sagemaker_training/smdataparallel.py
@@ -33,7 +33,8 @@
 
     # list of exceptions SMDDP wants training toolkit to catch and log
     exception_classes = [x for x in dir(exceptions) if isclass(getattr(exceptions, x))]
-except ImportError:
+# relaxed exception type in case of custom exceptions thrown during import
+except Exception:
     logger.info("No exception classes found in smdistributed.dataparallel")
     exception_classes = [errors.ExecuteUserScriptError]