johanels/photoprism
1
0
Fork 0
mirror of https://github.com/photoprism/photoprism.git synced 2025-12-12 08:44:04 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fixed classify and nsfw tests

Browse Source 
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
Download Patch File Download Diff File Expand all files Collapse all files

2
internal/ai/classify/model_test.go
View File

@@ -224,7 +224,7 @@ func TestModel_LoadModel(t *testing.T) {
err := tensorFlow.loadModel() err := tensorFlow.loadModel()
if err != nil { 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) assert.Error(t, err)

2
internal/ai/nsfw/nsfw_test.go
View File

@@ -13,7 +13,7 @@ import (
var modelPath, _ = filepath.Abs("../../../assets/nsfw") var modelPath, _ = filepath.Abs("../../../assets/nsfw")
var detector = NewModel(modelPath, 224, nil, false) var detector = NewModel(modelPath, nil, false)
func TestIsSafe(t *testing.T) { func TestIsSafe(t *testing.T) {
detect := func(filename string) Result { detect := func(filename string) Result {