fix: composite key unzip

Signed-off-by: Avik Basu <[email protected]>
numaproj · Nov 21, 2023 · b761c94 · b761c94
1 parent 9d1b999
commit b761c94
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/numalogic/udfs/postprocess.py b/numalogic/udfs/postprocess.py
@@ -137,9 +137,10 @@ def exec(self, keys: list[str], datum: Datum) -> Messages:
  # Forward payload if a training request is tagged
  if payload.header == Header.TRAIN_REQUEST or payload.status == Status.ARTIFACT_STALE:
  _conf = self.get_conf(payload.config_id)
+ ckeys = [_item[1] for _item in zip(_conf.composite_keys, payload.composite_keys)]
  train_payload = TrainerPayload(
  uuid=payload.uuid,
- composite_keys=_conf.composite_keys,
+ composite_keys=ckeys,
  metrics=payload.metrics,
  config_id=payload.config_id,
  )

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