Fix test

HumanCompatibleAI · Aug 25, 2024 · 390b946 · 390b946
1 parent 747e5b9
commit 390b946
Showing 1 changed file with 6 additions and 8 deletions.
diff --git a/perspective_ranker_test.py b/perspective_ranker_test.py
@@ -127,11 +127,9 @@ def test_arm_sort():
 
     result = rank.arm_sort(perspective_ranker.perspective_toxicity, scored_statements)
 
-    assert result == {
-        "ranked_ids": [
-            "test_statement_id_1",
-            "test_statement_id_unscorable",
-            "test_statement_id_2",
-            "test_statement_id_3",
-        ]
-    }
+    assert result["ranked_ids"] == [
+        "test_statement_id_1",
+        "test_statement_id_unscorable",
+        "test_statement_id_2",
+        "test_statement_id_3",
+    ]