Fix: cli/sched: use mi.Cpu class for CPU IDs

Signed-off-by: Philippe Proulx <eeppeliteloop@gmail.com>

diff --git a/lttnganalyses/cli/sched.py b/lttnganalyses/cli/sched.py
index 84a4aa8d42389a38dbcd8df3cfe3725143b30642..7e8ca18383093910cc33935d81e0edb2e6626dc2 100644 (file)
--- a/lttnganalyses/cli/sched.py
+++ b/lttnganalyses/cli/sched.py
@@ -64,7 +64,7 @@ class SchedAnalysisCommand(Command):
                 ('switch_ts', 'Switch timestamp', mi.Timestamp),
                 ('latency', 'Scheduling latency', mi.Duration),
                 ('prio', 'Priority', mi.Integer),
-                ('target_cpu', 'Target CPU', mi.Integer),
+                ('target_cpu', 'Target CPU', mi.Cpu),
                 ('wakee_proc', 'Wakee process', mi.Process),
                 ('waker_proc', 'Waker process', mi.Process),
             ]
@@ -76,7 +76,7 @@ class SchedAnalysisCommand(Command):
                 ('switch_ts', 'Switch timestamp', mi.Timestamp),
                 ('latency', 'Scheduling latency', mi.Duration),
                 ('prio', 'Priority', mi.Integer),
-                ('target_cpu', 'Target CPU', mi.Integer),
+                ('target_cpu', 'Target CPU', mi.Cpu),
                 ('wakee_proc', 'Wakee process', mi.Process),
                 ('waker_proc', 'Waker process', mi.Process),
             ]
@@ -330,7 +330,7 @@ class SchedAnalysisCommand(Command):
                 switch_ts=mi.Timestamp(sched_event.switch_ts),
                 latency=mi.Duration(sched_event.latency),
                 prio=mi.Integer(sched_event.prio),
-                target_cpu=mi.Integer(sched_event.target_cpu),
+                target_cpu=mi.Cpu(sched_event.target_cpu),
                 wakee_proc=wakee_proc,
                 waker_proc=waker_proc,
             )
@@ -363,7 +363,7 @@ class SchedAnalysisCommand(Command):
                 switch_ts=mi.Timestamp(sched_event.switch_ts),
                 latency=mi.Duration(sched_event.latency),
                 prio=mi.Integer(sched_event.prio),
-                target_cpu=mi.Integer(sched_event.target_cpu),
+                target_cpu=mi.Cpu(sched_event.target_cpu),
                 wakee_proc=wakee_proc,
                 waker_proc=waker_proc,
             )
@@ -692,7 +692,7 @@ class SchedAnalysisCommand(Command):
             switch_ts = row.switch_ts.value
             latency = row.latency.value
             prio = row.prio.value
-            target_cpu = row.target_cpu.value
+            target_cpu = row.target_cpu.id
             wakee_proc = row.wakee_proc
             waker_proc = row.waker_proc