From f8be5f54440d000b746ff4f5abebc999bcede3a8 Mon Sep 17 00:00:00 2001 From: RinZ27 <222222878+RinZ27@users.noreply.github.com> Date: Fri, 27 Mar 2026 19:54:03 +0700 Subject: [PATCH] Improve robustness in load_module by specifying Exception type --- tensorflow_quantum/core/ops/load_module.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tensorflow_quantum/core/ops/load_module.py b/tensorflow_quantum/core/ops/load_module.py index a98a27422..6fa350523 100644 --- a/tensorflow_quantum/core/ops/load_module.py +++ b/tensorflow_quantum/core/ops/load_module.py @@ -18,6 +18,7 @@ import types from distutils.sysconfig import get_python_lib +from tensorflow.python.framework import errors from tensorflow.python.framework import load_library from tensorflow.python.platform import resource_loader @@ -47,7 +48,7 @@ def _load(self): try: path = resource_loader.get_path_to_datafile(self._name) self._module = load_library.load_op_library(path) - except: + except (IOError, errors.NotFoundError): path = os.path.join(get_python_lib(), "tensorflow_quantum/core/ops", self._name) self._module = load_library.load_op_library(path)