From 8b98ecad70c0f26d4cd43e41c5d96adad9cf2251 Mon Sep 17 00:00:00 2001 From: Baiju Meswani Date: Thu, 31 Aug 2023 18:56:40 -0700 Subject: [PATCH] Change RuntimeError to ImportError (#17380) The `onnxruntime-validation` for ORTModule checks for `ImportError`: https://github.com/microsoft/onnxruntime/blob/44101e877125eaa18e191793973a4e1a002c6eca/onnxruntime/python/onnxruntime_validation.py#L73-L75 If any other kind of error is raised, it does not silently fail and will raise an exception. This causes a problem when ortmodule is explicitly not made available on win/mac packages since we currently raise a RuntimeError. Resolves issue: https://github.com/microsoft/onnxruntime-training-examples/issues/161 --- orttraining/orttraining/python/training/ortmodule/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/orttraining/orttraining/python/training/ortmodule/__init__.py b/orttraining/orttraining/python/training/ortmodule/__init__.py index 150f41eaec..59cf05bb08 100644 --- a/orttraining/orttraining/python/training/ortmodule/__init__.py +++ b/orttraining/orttraining/python/training/ortmodule/__init__.py @@ -18,7 +18,7 @@ from ._fallback import ORTModuleFallbackException, ORTModuleInitException, _Fall from .torch_cpp_extensions import is_installed as is_torch_cpp_extensions_installed if not is_ortmodule_available(): - raise RuntimeError("ORTModule is not supported on this platform.") + raise ImportError("ORTModule is not supported on this platform.") def _defined_from_envvar(name, default_value, warn=True):