fix: pipeline id key error

Signed-off-by: Avik Basu <[email protected]>
numaproj · Jan 23, 2024 · add642c · add642c
1 parent dc35c80
commit add642c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/numalogic/udfs/tools.py b/numalogic/udfs/tools.py
@@ -49,7 +49,7 @@ def get_df(
  -------
  dataframe and timestamps
  """
- _conf = stream_conf.ml_pipelines[data_payload["pipeline_id"]]
+ _conf = stream_conf.ml_pipelines[data_payload.get("pipeline_id", "default")]
  features = _conf.metrics
  df = (
  pd.DataFrame(data_payload["data"], columns=["timestamp", *features])