fix: conditional fwd tags

Signed-off-by: Avik Basu <ab93@users.noreply.github.com>
numaproj · Feb 28, 2024 · db943a6 · db943a6
1 parent 76b739c
commit db943a6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/numalogic/udfs/inference.py b/numalogic/udfs/inference.py
@@ -172,7 +172,7 @@ def exec(self, keys: list[str], datum: Datum) -> Messages:
             payload.metrics,
         )
         _increment_counter(counter=MSG_PROCESSED_COUNTER, labels=_metric_label_values)
-        msgs.append(Message(keys=keys, value=payload.to_json()))
+        msgs.append(Message(keys=keys, value=payload.to_json(), tags=["postprocess"]))
 
         _LOGGER.debug(
             "%s - Time taken in inference: %.4f sec",