From 795ce90b89eb38ab0c34681515286c39ceb6d998 Mon Sep 17 00:00:00 2001 From: Daara Shaw Date: Wed, 16 Oct 2024 18:27:54 +0100 Subject: [PATCH] Check `Library/bin` path exists before adding dll directory --- python/treelite/core.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/python/treelite/core.py b/python/treelite/core.py index 9c8ddac4..0393e76b 100644 --- a/python/treelite/core.py +++ b/python/treelite/core.py @@ -33,9 +33,10 @@ def _load_lib(): return None # type: ignore if sys.version_info >= (3, 8) and sys.platform == "win32": # pylint: disable=no-member - os.add_dll_directory( - os.path.join(os.path.normpath(sys.base_prefix), "Library", "bin") - ) + lib_bin_path = os.path.join(os.path.normpath(sys.base_prefix), "Library", "bin") + if os.path.isdir(lib_bin_path): + os.add_dll_directory(lib_bin_path) + lib = ctypes.cdll.LoadLibrary(lib_path[0]) lib.TreeliteGetLastError.restype = ctypes.c_char_p lib.log_callback = _log_callback