test: fix test

argilla-io · Apr 25, 2022 · b65acbb · b65acbb
1 parent 7841e46
commit b65acbb
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/labeling/text_classification/test_label_errors.py b/tests/labeling/text_classification/test_label_errors.py
@@ -157,7 +157,9 @@ def mock_find_label_issues(s, psx, n_jobs, **kwargs):
             assert kwargs == {
                 "mock": "mock",
                 "multi_label": is_multi_label,
-                "return_indices_ranked_by": "normalized_margin",
+                "return_indices_ranked_by": "normalized_margin"
+                if not is_multi_label
+                else "self_confidence",
             }
             return []