catch more issues

rapidsai · May 8, 2024 · 5c90c8e · 5c90c8e
1 parent 5046584
commit 5c90c8e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ucp/__init__.py b/ucp/__init__.py
@@ -21,7 +21,7 @@
 # Otherwise, we assume that the library was installed in a system path that ld can find.
 try:
     import libucx
-except ImportError:
+except (ImportError, OSError, RuntimeError):
     pass
 else:
     libucx.load_library()