diff --git a/numalogic/connectors/druid.py b/numalogic/connectors/druid.py index 9e329259..345a3a37 100644 --- a/numalogic/connectors/druid.py +++ b/numalogic/connectors/druid.py @@ -5,6 +5,7 @@ import pandas as pd import pytz from pydruid.client import PyDruid +from pydruid.utils.dimensions import DimensionSpec from pydruid.utils.filters import Filter from numalogic.connectors._config import Pivot from typing import Optional @@ -50,13 +51,15 @@ def fetch_data( start_dt = end_dt - timedelta(hours=hours) intervals = f"{start_dt.isoformat()}/{end_dt.isoformat()}" + dimension_specs = map(lambda d: DimensionSpec(dimension=d, output_name=d), dimensions) + params = { "datasource": datasource, "granularity": granularity, "intervals": intervals, "aggregations": aggregations, "filter": _filter, - "dimensions": dimensions, + "dimensions": dimension_specs, } _LOGGER.debug(