fix: remove rmw_take column

tier4 · Jan 7, 2025 · 3ca953c · 3ca953c
1 parent c4cd270
commit 3ca953c
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/caret_analyze/infra/lttng/records_provider_lttng.py b/src/caret_analyze/infra/lttng/records_provider_lttng.py
@@ -1379,9 +1379,11 @@ def fill_source_timestamp_with_latest_timestamp(records):
             f'{self._node_path.publish_topic_name}/rclcpp_publish_timestamp',
         ]
         left_key = sub_records.columns[0]
-        if COLUMN_NAME.RMW_TAKE_TIMESTAMP in columns:
-            columns.remove(COLUMN_NAME.RMW_TAKE_TIMESTAMP)
-            left_key = COLUMN_NAME.RMW_TAKE_TIMESTAMP
+        for column in sub_records.columns:
+            if column.endswith(COLUMN_NAME.RMW_TAKE_TIMESTAMP):
+                columns.remove(column)
+                left_key = column
+                break
 
         pub_sub_records = merge_sequential(
             left_records=sub_records,