fix: update test_tp_ranking_gr to match progress() 3-value return

progress() now returns (loss, global_tokens, extras) instead of
(loss, extras). Update all 3 call sites in the TP test.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Author: 2 people

examples/hstu/test/tensor_parallel/test_tp_ranking_gr.py

@@ -278,11 +278,15 @@ def test_tp_gr_ranking_forward_backward_update(
             tp_ranking_gr, debug_ranking_gr, debug_ranking_gr_fp32
         )
         for i, batch in enumerate(history_batches):
-            _, (losses, logits, _, _) = debug_pipeline.progress(debug_pipeline_batches)
-            _, (losses_fp32, logits_fp32, _, _) = debug_pipeline_fp32.progress(
+            _, _, (losses, logits, _, _) = debug_pipeline.progress(
+                debug_pipeline_batches
+            )
+            _, _, (losses_fp32, logits_fp32, _, _) = debug_pipeline_fp32.progress(
                 debug_pipeline_batches_fp32
             )
-            _, (tp_losses, tp_logits, _, _) = tp_pipeline.progress(iter_history_batches)
+            _, _, (tp_losses, tp_logits, _, _) = tp_pipeline.progress(
+                iter_history_batches
+            )
             torch.distributed.barrier(device_ids=[torch.cuda.current_device()])
             compare_tpN_to_debug_weights(
                 tp_ranking_gr, debug_ranking_gr, debug_ranking_gr_fp32