Skip to content

Commit 983f0f2

Browse files
committed
test: update tests
1 parent 732dfee commit 983f0f2

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

tests/modules/retrieval/test_logreg.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -36,13 +36,9 @@ def test_score_evaluates_model():
3636
mock_context.data_handler.test_utterances.return_value = ["hello", "goodbye"]
3737
mock_context.data_handler.test_labels.return_value = [0, 1]
3838

39-
def mock_metric_fn(true_labels, predicted_labels):
40-
return sum(1 for t, p in zip(true_labels, predicted_labels[0], strict=False) if t == p) / len(true_labels)
39+
scores = module.score(mock_context, split="test")
4140

42-
score = module.score(mock_context, split="test", metric_fn=mock_metric_fn)
43-
44-
assert 0 <= score <= 1
45-
assert score > 0
41+
assert isinstance(scores, dict)
4642

4743

4844
def test_dump_and_load_preserves_model_state():

0 commit comments

Comments
 (0)