From 230be5bbb4a1ecb588396af1bcf7365066a6d508 Mon Sep 17 00:00:00 2001 From: Douglas Raillard Date: Mon, 15 Jul 2024 14:53:04 +0100 Subject: [PATCH] lisa.trace: Avoid errors when dropping non-existent columns FIX --- lisa/trace.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lisa/trace.py b/lisa/trace.py index 3bfdbe340..d19e8a619 100644 --- a/lisa/trace.py +++ b/lisa/trace.py @@ -1020,7 +1020,7 @@ class TraceDumpTraceParser(TraceParserBase): pl.col('Time').cast(pl.Duration("ns")), pl.col('__pid').replace_strict(pid_comms, default=None).alias('__comm') ]) - df = df.drop('common_type', 'common_flags', 'common_preempt_count') + df = df.drop(('common_type', 'common_flags', 'common_preempt_count'), strict=False) monotonic_clocks = ( 'local', @@ -1455,7 +1455,7 @@ class TxtTraceParserBase(TraceParserBase): # We only needed the fields to infer the descriptors, so let's drop # them to lower peak memory usage with contextlib.suppress(KeyError): - self._skeleton_df = self._skeleton_df.drop(['__fields']) + self._skeleton_df = self._skeleton_df.drop(['__fields'], strict=False) event_parsers = { **{ -- GitLab