updated non-existing int values to -1

hpcgroup · Nov 8, 2024 · 27e95f8 · 27e95f8
1 parent 02a00e5
commit 27e95f8
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/pipit/readers/core_reader.py b/pipit/readers/core_reader.py
@@ -93,15 +93,19 @@ def finalize(self):
         # create a dataframe
         trace_df = pandas.DataFrame(all_events)
 
+        trace_df["_matching_event"].fillna(-1, inplace=True)
+        trace_df["_parent"].fillna(-1, inplace=True)
+        trace_df["_matching_timestamp"].fillna(-1, inplace=True)
+
         # categorical for memory savings
         trace_df = trace_df.astype(
             {
                 "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
@@ -115,7 +119,7 @@ def __update_parent_child_relationships(
         """
         if len(stack) == 0:
             # root event
-            event["_parent"] = numpy.nan
+            event["_parent"] = -1
         else:
             parent_event = event_list[stack[-1]]
             event["_parent"] = parent_event["unique_id"]