Fix device order.

src/visualizer.py

@@ -243,7 +243,10 @@ def create_pipeline_figure(schedule_data: Dict[int, List[Dict]], max_time=None,
 
     # Set axis properties
     device_labels = [f"Device {i}" for i in range(num_devices)]
-    device_labels.reverse()  # Reverse to put Device 0 at the top
+    # Modify the ordering to put Device 1 at the top, then Device 0, then the rest
+    if num_devices >= 2:
+        # Move Device 1 to the top, followed by Device 0
+        device_labels = [device_labels[1], device_labels[0]] + device_labels[2:] if num_devices > 1 else device_labels
     
     # Calculate tick positions with no gaps
     tick_positions = [(num_devices - i - 1) * y_spacing for i in range(num_devices)]