Skip to content

Commit 6251632

Browse files
committed
fix(test): update MockPIIInitializer signature to match interface
Update MockPIIInitializer.Init() to include numClasses parameter to match the PIIInitializer interface changes. This fixes the CI test failure where the mock didn't properly implement the updated interface signature. Signed-off-by: Yossi Ovadia <[email protected]>
1 parent cf52321 commit 6251632

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/semantic-router/pkg/classification/classifier_test.go

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -718,7 +718,9 @@ var _ = Describe("jailbreak detection", func() {
718718

719719
type MockPIIInitializer struct{ InitError error }
720720

721-
func (m *MockPIIInitializer) Init(_ string, useCPU bool) error { return m.InitError }
721+
func (m *MockPIIInitializer) Init(_ string, useCPU bool, numClasses int) error {
722+
return m.InitError
723+
}
722724

723725
type MockPIIInferenceResponse struct {
724726
classifyTokensResult candle_binding.TokenClassificationResult

0 commit comments

Comments
 (0)