fix: add threshold in metadata

Signed-off-by: s0nicboOm <[email protected]>
numaproj · Apr 29, 2024 · f9ebfaf · f9ebfaf
1 parent 6e41a79
commit f9ebfaf
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/numalogic/udfs/postprocess.py b/numalogic/udfs/postprocess.py
@@ -146,6 +146,13 @@ def exec(self, keys: list[str], datum: Datum) -> Messages:
  score_conf=_conf.numalogic_conf.score,
  postproc_tx=postproc_tx,
  ) # (nfeat,)
+ payload = replace(
+ payload,
+ metadata={
+ "threshold": float(thresh_artifact.artifact.threshold),
+ **payload.metadata,
+ },
+ )
 
  # Calculate adjusted unified score
  a_adjusted, y_unified, y_features = self._adjust_score(_conf, a_unified, payload)

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