fix: send conf keys instead of request keys to trainer

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

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/numalogic/udfs/postprocess.py b/numalogic/udfs/postprocess.py
@@ -136,9 +136,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)
  train_payload = TrainerPayload(
  uuid=payload.uuid,
- composite_keys=keys,
+ composite_keys=_conf.composite_keys,
  metrics=payload.metrics,
  config_id=payload.config_id,
  )

diff --git a/numalogic/udfs/trainer/_prom.py b/numalogic/udfs/trainer/_prom.py
@@ -80,5 +80,4 @@ def fetch_data(self, payload: TrainerPayload) -> pd.DataFrame:
  time.perf_counter() - _start_time,
  _df.shape,
  )
- _df.to_csv(f"{payload.config_id}.csv")
  return _df
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "numalogic"
-version = "0.6.1.dev0"
+version = "0.6.1.dev1"
 description = "Collection of operational Machine Learning models and tools."
 authors = ["Numalogic Developers"]
 packages = [{ include = "numalogic" }]