Skip to content

Commit 51bbc9b

Browse files
authored
fix tests (#1134)
1 parent 7f081ac commit 51bbc9b

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

tests/modelgauge_tests/test_together_sut_factory.py renamed to tests/modelgauge_tests/sut_tests/test_together_sut_factory.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010

1111

1212
def test_make_sut():
13-
with patch("modelgauge.suts.together_sut_factory.TogetherSUTFactory.find", return_value="google/gemma:together"):
13+
with patch("modelgauge.suts.together_sut_factory.TogetherSUTFactory._find", return_value="google/gemma:together"):
1414
sut_metadata = DynamicSUTMetadata(model="gemma", maker="google", driver="together")
1515
found_sut = TogetherSUTFactory.make_sut(sut_metadata)
1616
assert found_sut is not None
@@ -21,7 +21,7 @@ def test_make_sut_bad_model():
2121
os.environ["TOGETHER_API_KEY"] = "fake-key"
2222

2323
sut_metadata = DynamicSUTMetadata(model="bogus", maker="fake", driver="together")
24-
with patch("modelgauge.suts.together_sut_factory.TogetherSUTFactory.find", side_effect=ModelNotSupportedError()):
24+
with patch("modelgauge.suts.together_sut_factory.TogetherSUTFactory._find", side_effect=ModelNotSupportedError()):
2525
with pytest.raises(ModelNotSupportedError):
2626
_ = TogetherSUTFactory.make_sut(sut_metadata)
2727

@@ -34,7 +34,7 @@ def test_find():
3434
return_value=[{"id": "google/gemma"}],
3535
):
3636
sut_metadata = DynamicSUTMetadata(model="gemma", maker="google", driver="together")
37-
assert TogetherSUTFactory.find(sut_metadata) == sut_metadata.external_model_name()
37+
assert TogetherSUTFactory._find(sut_metadata) == sut_metadata.external_model_name()
3838

3939

4040
def test_find_bad_model():
@@ -44,4 +44,4 @@ def test_find_bad_model():
4444
return_value=None,
4545
):
4646
with pytest.raises(ModelNotSupportedError):
47-
_ = TogetherSUTFactory.find(sut_metadata)
47+
_ = TogetherSUTFactory._find(sut_metadata)

0 commit comments

Comments
 (0)