Skip to content

Commit 48c7e39

Browse files
committed
fix: fixed split in score
1 parent 6ce853e commit 48c7e39

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/modules/retrieval/test_logreg.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,8 +30,8 @@ def test_score_evaluates_model():
3030

3131
mock_context = MagicMock()
3232
mock_context.data_handler.test_utterances.return_value = ["hello", "goodbye"]
33-
mock_context.data_handler.test_labels.return_value = [0, 1]
33+
mock_context.data_handler.test_labels.return_value = [[1, 0], [0, 1]]
3434

35-
scores = module.score(mock_context, split="validation")
35+
scores = module.score(mock_context, split="test")
3636

3737
assert isinstance(scores, dict)

0 commit comments

Comments
 (0)