Fix: Changing pivot to pivot_table to support aggregation (#376)

Explain what this PR does. Changing pivot to pivot_table to support aggregation --------- Signed-off-by: Nandita Koppisetty <[email protected]>
numaproj · May 4, 2024 · 402aabf · 402aabf
1 parent 141e9a0
commit 402aabf
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/numalogic/connectors/druid/_druid.py b/numalogic/connectors/druid/_druid.py
@@ -199,10 +199,11 @@ def fetch(
  if pivot:
  pivoted_frames = []
  for column in pivot.columns:
- _df = df.pivot(
+ _df = df.pivot_table(
  index=pivot.index,
  columns=[column],
  values=pivot.value,
+ aggfunc="sum"
  )
  pivoted_frames.append(_df)
 
@@ -304,10 +305,11 @@ def chunked_fetch(
  if pivot:
  pivoted_frames = []
  for column in pivot.columns:
- _df = df.pivot(
+ _df = df.pivot_table(
  index=pivot.index,
  columns=[column],
  values=pivot.value,
+ aggfunc="sum"
  )
  pivoted_frames.append(_df)
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "numalogic"
-version = "0.9.1a9"
+version = "0.9.1a10"
 description = "Collection of operational Machine Learning models and tools."
 authors = ["Numalogic Developers"]
 packages = [{ include = "numalogic" }]