Skip to content
9 changes: 5 additions & 4 deletions pipit/readers/core_reader.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
from typing import List, Dict

import pandas
import numpy as np
from pipit.trace import Trace


Expand Down Expand Up @@ -102,9 +103,9 @@ def finalize(self):
"Name": "category",
"Event Type": "category",
"Process": "category",
"_matching_event": "int32",
"_parent": "int32",
"_matching_timestamp": "int32",
"_matching_event": "Int32",
"_parent": "Int32",
"_matching_timestamp": "Int32",
}
)
return trace_df
Expand All @@ -118,7 +119,7 @@ def __update_parent_child_relationships(
"""
if len(stack) == 0:
# root event
event["_parent"] = -1
event["_parent"] = np.nan
else:
parent_event = event_list[stack[-1]]
event["_parent"] = parent_event["unique_id"]
Expand Down
Loading
Loading