diff --git a/tools/gpu_assignment.py b/tools/gpu_assignment.py
index f300b888..8406627c 100644
--- a/tools/gpu_assignment.py
+++ b/tools/gpu_assignment.py
@@ -346,7 +346,7 @@ def _split_lps_cost(self, assignment: Assignment) -> float:
     def best_swap_for(self, slot_a: GpuSlot, assignment, current_cost, slot_subset=None):
         slot_subset = self.gpu_slots if slot_subset is None else slot_subset
         costs = [(current_cost, slot_a)]
-        for i, slot_b in enumerate(tqdm(slot_subset, desc='best_swap_for', leave=False)):
+        for i, slot_b in enumerate(slot_subset):
             if slot_a.node == slot_b.node and slot_a.gpu == slot_b.gpu:
                 # No point swapping pairs already on the same device
                 continue