fix: thresh bug when there is RedisRegistryError (#164)

Signed-off-by: kbatra <[email protected]>
numaproj · Jul 31, 2023 · 5653128 · 5653128
1 parent 1eafb0d
commit 5653128
Show file tree

Hide file tree

Showing 3 changed files with 267 additions and 202 deletions.
diff --git a/numaprom/udf/threshold.py b/numaprom/udf/threshold.py
@@ -80,7 +80,10 @@ def threshold(_: list[str], datum: Datum) -> list[tuple[str, bytes]]:
         )
         payload.set_header(Header.STATIC_INFERENCE)
         payload.set_status(Status.RUNTIME_ERROR)
-        return orjson.dumps(payload, option=orjson.OPT_SERIALIZE_NUMPY)
+        return [
+            (TRAIN_VTX_KEY, orjson.dumps(train_payload)),
+            (POSTPROC_VTX_KEY, _get_static_thresh_payload(payload, metric_config)),
+        ]
     except Exception as ex:
         LOGGER.exception(
             "{uuid} - Unhandled exception while fetching threshold artifact, "