1010
1111
1212def 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
4040def 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