fix pytest

Signed-off-by: Koudai Yamasaki <[email protected]>
tier4 · Jan 23, 2025 · e65a8f9 · e65a8f9
1 parent b99f36b
commit e65a8f9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src/caret_analyze/runtime/path.py b/src/caret_analyze/runtime/path.py
@@ -165,7 +165,8 @@ def is_match_column(column: str, target_name: str) -> bool:
 
             # remove columns from included in communications not via callback
             if is_match_column(right_records.columns[-1], 'callback_start_timestamp') and \
-                    all(x is None for x in right_records.get_column_series(right_records.columns[-1])):
+                    all(x is None for x
+                        in right_records.get_column_series(right_records.columns[-1])):
                 right_records.drop_columns([right_records.columns[-1]])
 
             if left_records.columns[-1] != right_records.columns[0]:

diff --git a/src/test/infra/lttng/test_latency_definitions.py b/src/test/infra/lttng/test_latency_definitions.py
@@ -1381,6 +1381,7 @@ def test_inter_proc_empty_data(
             columns=[
                 f'{communication.topic_name}/rclcpp_publish_timestamp',
                 f'{communication.topic_name}/source_timestamp',
+                f'{callback.callback_name}/callback_start_timestamp',
             ],
             dtype='Int64'
         )