handle not installed import error

getsentry · Dec 22, 2024 · 38f279b · 38f279b
1 parent 4ee3667
commit 38f279b
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/sentry_sdk/integrations/unleash.py b/sentry_sdk/integrations/unleash.py
@@ -3,9 +3,12 @@
 
 import sentry_sdk
 from sentry_sdk.flag_utils import flag_error_processor
-from sentry_sdk.integrations import Integration
+from sentry_sdk.integrations import Integration, DidNotEnable
 
-from UnleashClient import UnleashClient
+try:
+    from UnleashClient import UnleashClient
+except ImportError:
+    raise DidNotEnable("UnleashClient is not installed")
 
 if TYPE_CHECKING:
     from typing import Any