From 2babbf310de586ba8a144e9d01f6fbb97c803624 Mon Sep 17 00:00:00 2001 From: Barbara Korycki Date: Mon, 13 Jan 2025 13:47:40 -0800 Subject: [PATCH] Fix tests --- tests/modelgauge_tests/test_default_annotator_set.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/tests/modelgauge_tests/test_default_annotator_set.py b/tests/modelgauge_tests/test_default_annotator_set.py index 2aa7c7e4..1fff6f59 100644 --- a/tests/modelgauge_tests/test_default_annotator_set.py +++ b/tests/modelgauge_tests/test_default_annotator_set.py @@ -1,16 +1,15 @@ import pytest from unittest.mock import MagicMock from modelgauge.default_annotator_set import DefaultAnnotatorSet -from modelgauge.suts.together_client import TogetherApiKey def test_constructor(): - annotators = DefaultAnnotatorSet(secrets={"together_api_key": TogetherApiKey("fake")}) + annotators = DefaultAnnotatorSet() assert len(annotators.annotators) == 1 assert "llama_guard_2" in annotators.annotators def test_evaluate(): - annotators = DefaultAnnotatorSet(secrets={"together_api_key": TogetherApiKey("fake")}) + annotators = DefaultAnnotatorSet() item = MagicMock() assert type(annotators.evaluate(item).get("is_safe", None)) == float