fix: druid opt dependency issue

Signed-off-by: Avik Basu <ab93@users.noreply.github.com>
numaproj · Sep 13, 2023 · 8d674f9 · 8d674f9
1 parent b3cd337
commit 8d674f9
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/numalogic/connectors/__init__.py b/numalogic/connectors/__init__.py
@@ -1,3 +1,5 @@
+from importlib.util import find_spec
+
 from numalogic.connectors._config import (
     RedisConf,
     PrometheusConf,
@@ -7,7 +9,6 @@
     ConnectorType,
 )
 from numalogic.connectors.prometheus import PrometheusFetcher
-from numalogic.connectors.druid import DruidFetcher
 
 __all__ = [
     "RedisConf",
@@ -17,5 +18,12 @@
     "DruidFetcherConf",
     "ConnectorType",
     "PrometheusFetcher",
-    "DruidFetcher",
 ]
+
+try:
+    find_spec("pydruid")
+except (ImportError, ModuleNotFoundError):
+    pass
+else:
+
+    __all__.append("DruidFetcher")