diff --git a/lisa/trace.py b/lisa/trace.py index 3bfdbe340ba825418e74dea9199199613ae572f0..d19e8a6194ab77feb8d9735e632cfb5812525427 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 = { **{