fix: druid fetcher config bug

Signed-off-by: nkoppisetty <nandita.iitkgp@gmail.com>
numaproj · Aug 23, 2023 · e50e354 · e50e354
1 parent ca76f85
commit e50e354
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/udf/anomaly-detection/src/connectors/_config.py b/udf/anomaly-detection/src/connectors/_config.py
@@ -40,7 +40,11 @@ class Pivot:
 class DruidFetcherConf:
     datasource: str
     dimensions: list[str] = field(default_factory=list)
-    aggregations: dict = field(default_factory=lambda: {"count": doublesum("count")})
+    aggregations: dict = field(default_factory=lambda: dict)
     group_by: list[str] = field(default_factory=list)
     pivot: Pivot = field(default_factory=lambda: Pivot())
     granularity: str = "minute"
+
+    def __post_init__(self):
+        if not self.aggregations:
+            self.aggregations = {"count": doublesum("count")}
diff --git a/udf/anomaly-detection/src/connectors/druid.py b/udf/anomaly-detection/src/connectors/druid.py
@@ -74,7 +74,7 @@ def fetch_data(
         if group_by:
             df = df.groupby(by=group_by).sum().reset_index()
 
-        if pivot:
+        if pivot.columns:
             df = df.pivot(
                 index=pivot.index,
                 columns=pivot.columns,