Fix applying seq_ids to start/stop cols

nv-morpheus · Oct 31, 2024 · 7b5f54a · 7b5f54a
1 parent e09deac
commit 7b5f54a
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/examples/log_parsing/postprocessing.py b/examples/log_parsing/postprocessing.py
@@ -87,9 +87,8 @@ def _postprocess(self, msg: ControlMessage):
             src_index = src_df.index.to_pandas()
 
         seq_ids = msg.tensors().get_tensor('seq_ids').get()
+        infer_pdf = pd.DataFrame({"doc": src_index, "start": seq_ids[:, 1], "stop": seq_ids[:, 2]})
 
-        infer_pdf = pd.DataFrame({"doc": src_index, "start": seq_ids[:, 0], "stop": seq_ids[:, 1]})
-        infer_pdf.columns = ["doc", "start", "stop"]
         infer_pdf["confidences"] = msg.tensors().get_tensor('confidences').tolist()
         infer_pdf["labels"] = msg.tensors().get_tensor('labels').tolist()
         infer_pdf["token_ids"] = msg.tensors().get_tensor('input_ids').tolist()