We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent bf17a11 commit 0151b2eCopy full SHA for 0151b2e
src/sagemaker_training/smdataparallel.py
@@ -33,7 +33,8 @@
33
34
# list of exceptions SMDDP wants training toolkit to catch and log
35
exception_classes = [x for x in dir(exceptions) if isclass(getattr(exceptions, x))]
36
-except ImportError:
+# relaxed exception type in case of custom exceptions thrown during import
37
+except Exception: # noqa W0703
38
logger.info("No exception classes found in smdistributed.dataparallel")
39
exception_classes = [errors.ExecuteUserScriptError]
40
0 commit comments