Skip to content

Commit 31e345b

Browse files
committed
fix: use tagged switch instead of if-else (staticcheck QF1003)
Convert if-else chain to switch statement in TestFactorizationMachines_WeightedTraining for cleaner code.
1 parent b254091 commit 31e345b

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

model/ctr/model_test.go

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -262,10 +262,11 @@ func TestFactorizationMachines_WeightedTraining(t *testing.T) {
262262
foundPurchase := false
263263
for i := 0; i < train.Count(); i++ {
264264
w := train.GetWeight(i)
265-
if train.FeedbackTypes[i] == "click" {
265+
switch train.FeedbackTypes[i] {
266+
case "click":
266267
assert.Equal(t, float32(1.0), w)
267268
foundClick = true
268-
} else if train.FeedbackTypes[i] == "purchase" {
269+
case "purchase":
269270
assert.Equal(t, float32(5.0), w)
270271
foundPurchase = true
271272
}

0 commit comments

Comments
 (0)