minor qb fix

pomonam · Mar 19, 2024 · 6e65e88 · 6e65e88
1 parent f6db06d
commit 6e65e88
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/gpu_tests/ddp_test.py b/tests/gpu_tests/ddp_test.py
@@ -210,7 +210,7 @@ def test_lr_pairwise_scores(self) -> None:
             score_args=score_args,
             overwrite_output_dir=True,
         )
-        new_pairwise_scores = self.analyzer.load_pairwise_scores(scores_name=NEW_SCORE_NAME)
+        new_pairwise_scores = self.analyzer.load_pairwise_scores(scores_name="ddp_qb")
 
         if LOCAL_RANK == 0:
             print(f"Previous score: {pairwise_scores[ALL_MODULE_NAME][0]}")