Fixed classify and nsfw tests

They were broken when the constructors changed to include metadata.
This commit is contained in:
raystlin
2025-04-13 14:53:34 +00:00
parent d993eb2a85
commit ad94d03576
2 changed files with 2 additions and 2 deletions

View File

@@ -224,7 +224,7 @@ func TestModel_LoadModel(t *testing.T) {
err := tensorFlow.loadModel()
if err != nil {
assert.Contains(t, err.Error(), "no such file or directory")
assert.Contains(t, err.Error(), "Could not find SavedModel")
}
assert.Error(t, err)